resume doesn't load 

Hello, it's great work! Really great.

However, the resume doen't work. 

Besides, in training, the optimizer is created on model's cpu, however, the model is later put on gpu by model.cuda() in train.py, it will cause many issues later. 
