def save_checkpoint(model,
optimizer,
epoch,
dataloaders,
checkpoint_dir,
use_thread=False):
dict_to_save = {
'state_dict': model.state_dict(),
'epoch': epoch,
'model': model,
'optimizer': optimizer,
'curr_iter_idx': dataloaders['train'].curr_iter_idx,
}
checkpoints.save_checkpoint(dict_to_save, checkpoint_dir, epoch)
return []
if __name__ == '__main__':
assert LOG_DIR, 'log dir cannot be empty'
# Manually parse command line opts
short_usage, usage, internal_usage = ex.get_usage()
args = docopt(internal_usage, [str(a) for a in sys.argv[1:]], help=False)
config_updates, named_configs = get_config_updates(args['UPDATE'])
ex.run('prologue', config_updates, named_configs, options=args)
ex.observers.append(FileStorageObserverWithExUuid.create(LOG_DIR))
ex.run_commandline()
else:
print(__name__)
