Plotting Error While Doing Continue Training

[hmza09]

Hi, I have been working on this repository for learning purposes. I found it helpful. After training on different models, I wanted to try the argument of Continue_Training. So I set the continue_training to True and epoch_start_i to 300. As I previously trained for 300 epochs and a total number of epochs to 310.
But, after one epoch, it shows some error in plotting in the arguments of:
ax1.plot(range(epoch+1), avg_scores_per_epoch)

valueError: x and y must have same first dimension, but have shapes (300,) and (1,)

I guess, it's because of avg score initialization from an empty array. Can you kindly look into it.

[nistarlwc]

same problem