Open hyunjik11 opened 4 years ago
Another minor issue: the run_epoch method in main.py is missing arguments: model, device, args, log_p_T, optimizer. Hence I think the lines where run_epoch is called in main should be modified accordingly.
run_epoch
model, device, args, log_p_T, optimizer
main
Another minor issue: the
run_epochmethod in main.py is missing arguments:model, device, args, log_p_T, optimizer. Hence I think the lines whererun_epochis called inmainshould be modified accordingly.