GeorgeSeif / Semantic-Segmentation-Suite

Semantic Segmentation Suite in TensorFlow. Implement, train, and test new Semantic Segmentation models easily!
2.5k stars 880 forks source link

Plotting Error While Doing Continue Training #245

Open hmza09 opened 3 years ago

hmza09 commented 3 years ago

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 commented 3 years ago

same problem