Fix uint8 issue for newer versions of PyTorch

philtabor / Deep-Q-Learning-Paper-To-Code

MIT License

351 stars 146 forks source link

Open srikanthkb opened 4 years ago

srikanthkb commented 4 years ago

uint8 is deprecated in recent versions of Pytorch(1.2 or higher), have updated the variable to tensor.bool()