facebookresearch/ELF

Can not run train_minirts.sh

csethomas opened this issue · 0 comments

When I execute the train_minirts.sh, I encounter the following errors:
Traceback (most recent call last):
File "train.py", line 14, in <module>
from rlpytorch import *
File "/home/github/ELF/rlpytorch/__init__.py", line 15, in <module>
from .trainer import Trainer, Evaluator, LSTMTrainer
File "/home/github/ELF/rlpytorch/trainer/__init__.py", line 8, in <module>
from .trainer import Trainer, Evaluator
File "/home/github/ELF/rlpytorch/trainer/trainer.py", line 10, in <module>
import utils_elf
File "/home/thomas/ELF/rlpytorch/trainer/../../elf/utils_elf.py", line 191
def transfer_cpu2gpu(self, batch_gpu, async=True):