buntyke/shepherd_gym

OpenAI Gym environment for dog-sheep shepherding task

shepherd_gym's Issues

Zero Division Error

Hello!!

I am trying to train from this code, but when i execute it, I have the following error:

Experiment: heuristic

Initialization Done

Data Loaded

Network Created

Traceback (most recent call last):
File "shepherd_imitation.py", line 210, in
main()
File "shepherd_imitation.py", line 151, in main
output = trainer.run()
File "/home/franci/shepherd_gym/shepherd_gym/models/imitation_utils.py", line 113, in run
self.train()
File "/home/franci/shepherd_gym/shepherd_gym/models/imitation_utils.py", line 176, in train
{'train':100.*correct/len(self.train_loader.dataset)},
ZeroDivisionError: float division by zero

Could you please help me to solve this problem? Does this occur to you too?

Best regards,
Franci