Problems with training the model

[ManHinnn0509]

Hi, I finally got everything ready and it was my first time training a model. However I noticed some problems with lofi2lofi_train.py:

1. Unable to continue training with pre-trained / previous results
   Everytime I run the script, it trains from epoch 0 and doesn't load any files. How make the script to load previous results and continue training?

2. High CPU / disk usage
   I notice that the RAM usage is always rising and extreme high. It reached MemoryError at 211 epochs.
   Is there anyway to improve the training script?

I'm not sure when it will start overfitting but for now, I want to have as much epochs as possible.

Really love this project and I want to make it better.

Thank you!