Could not reproduce pretrained model

[jpsml]

I have followed the instructions for downloading and preparing training data and tried to train a model by changing load_model to False and both save_model and train_model to True in DCI-training-0.0.2.py. I also needed to change two lines because I am using pytorch 1.0.0.

I have changed this line:
acc_loss +=np.sum(loss.data[0])

to this:
acc_loss +=np.sum(loss.item())

And I have changed this line:
val_loss +=np.sum(loss.data[0])

to this:
val_loss +=np.sum(loss.item())

However, I am getting the following results:

The final values for loss and validation loss were 2.2964887819458952 and 2.3019133200951383, respectively. Is this expected? Should I change any other parameter (like number of epochs) in order to reproduce the pretrained model?

I am using Ubuntu 18.04.1 LTS, Python 2.7.15rc1 and pandas 0.23.4.

Best regards.