`run_epoch` in main.py is missing arguments

[hyunjik11]

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.