Hyunjik Kim

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