Keep track of epoch state during training

Have a global variable in the base experiment class called `epoch_num`. Child classes can call this whenever they'd like. Also, on the callbacks, pass this information in like so: 

```python
def on_epoch_end(self, epoch):
        ... 
```