vincentherrmann / pytorch-wavenet

An implementation of WaveNet with fast generation
MIT License
968 stars 225 forks source link

ZeroDivisionError: division by zero #41

Open Sususuqii opened 3 years ago

Sususuqii commented 3 years ago

Traceback (most recent call last): File "train_script.py", line 87, in continue_training_at_step=0) File "/disk2/sushunqi/pytorch-wavenet-master/wavenet_training.py", line 90, in train self.logger.log(step, loss) File "/disk2/sushunqi/pytorch-wavenet-master/model_logging.py", line 35, in log self.validate(current_step) File "/disk2/sushunqi/pytorch-wavenet-master/model_logging.py", line 86, in validate avg_loss, avg_accuracy = self.trainer.validate() File "/disk2/sushunqi/pytorch-wavenet-master/wavenet_training.py", line 110, in validate avg_loss = total_loss / len(self.dataloader) ZeroDivisionError: division by zero

Thank you very much!!