浏览代码

_

/exp-vince
vincentpierre 4 年前
当前提交
a899ecff
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 2
      ml-agents/mlagents/torch_utils/torch.py

2
ml-agents/mlagents/torch_utils/torch.py


# Known PyLint compatibility with PyTorch https://github.com/pytorch/pytorch/issues/701
# pylint: disable=E1101
if torch.cuda.is_available():
if False:#torch.cuda.is_available():
torch.set_default_tensor_type(torch.cuda.FloatTensor)
device = torch.device("cuda")
else:

正在加载...
取消
保存