Keep track of epoch state during training

[rosikand]

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:

def on_epoch_end(self, epoch):
        ...