Cuda valueError - Githubissues

beta-team / beta-recsys

Beta-RecSys: Build, Evaluate and Tune Automated Recommender Systems

https://beta-recsys.readthedocs.io/en/latest/

MIT License

162 stars 33 forks source link

Cuda valueError #343

Closed Yingting-dev closed 3 years ago

Yingting-dev commented 3 years ago

Describe the bug

in train_engine.py line 55 int(self.config["system"]["device"].replace("cuda", "")), ValueError: invalid literal for int() with base 10: ''

Describe your attempts

works by int(self.config["system"]["device"].replace("cuda", "0")),

Context

OS [Windows 10]:
Hardware [GTX 1060]:
Environment [Pytorch 1.4, Tensorflow 2.0]:

Additional Information