You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
32 lines
499 B
32 lines
499 B
def before_train(trainer): |
|
# Initialize tensorboard logger |
|
pass |
|
|
|
|
|
def on_epoch_start(trainer): |
|
pass |
|
|
|
|
|
def on_batch_start(trainer): |
|
pass |
|
|
|
|
|
def on_val_start(trainer): |
|
pass |
|
|
|
|
|
def on_val_end(trainer): |
|
pass |
|
|
|
|
|
def on_model_save(trainer): |
|
pass |
|
|
|
|
|
default_callbacks = { |
|
"before_train": before_train, |
|
"on_epoch_start": on_epoch_start, |
|
"on_batch_start": on_batch_start, |
|
"on_val_start": on_val_start, |
|
"on_val_end": on_val_end, |
|
"on_model_save": on_model_save}
|
|
|