konradheimpel
Results
2
issues of
konradheimpel
Hi, trainer.py has lots of cuda() calls without asking about the CUDA flag set in config file, so evaluation/training on a CPU is not possible. I fixed the issue by...
Added "if CUDA" to all cuda() calls in trainer.py Makes it possible to evaluate & train on CPU