Giters

kaylode / theseus

General template for most Pytorch projects

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Implement independent Callbacks class

kaylode opened this issue · comments

commented

For example:

Source: tensorflow - keras

class CustomCallback(keras.callbacks.Callback):
    def on_train_begin(self, logs=None):
        keys = list(logs.keys())
        print("Starting training; got log keys: {}".format(keys))

    def on_train_end(self, logs=None):
        keys = list(logs.keys())
        print("Stop training; got log keys: {}".format(keys))

    def on_epoch_begin(self, epoch, logs=None):
        keys = list(logs.keys())
        print("Start epoch {} of training; got log keys: {}".format(epoch, keys))

    def on_epoch_end(self, epoch, logs=None):
        keys = list(logs.keys())
        print("End epoch {} of training; got log keys: {}".format(epoch, keys))
  • Example usage:
trainer = ClassificationTrainer(..., callbacks = List[Callbacks])