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_epoch
method in main.py is missing arguments:model, device, args, log_p_T, optimizer
. Hence I think the lines whererun_epoch
is called inmain
should be modified accordingly.