`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.