Fix uint8 issue for newer versions of PyTorch

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

MIT License

342 stars 145 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()