weidler / RLaSpa

Reinforcement Learning in Latent Space

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Verify if GPU is indeed utilized

dannigt opened this issue · comments

Verify whether if torch.cuda.is_available(): torch.set_default_tensor_type('torch.cuda.FloatTensor') doing what's intended