Reinforcement Learning in Latent Space
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool
dannigt opened this issue 6 years ago · comments
Verify whether if torch.cuda.is_available(): torch.set_default_tensor_type('torch.cuda.FloatTensor') doing what's intended
if torch.cuda.is_available(): torch.set_default_tensor_type('torch.cuda.FloatTensor')