ys7yoo / claf

CLaF: Open-Source Clova Language Framework
https://naver.github.io/claf/
MIT License
0 stars 0 forks source link

GPU error during prediction #9

Open ys7yoo opened 3 years ago

ys7yoo commented 3 years ago
/home/yyoo/torch/lib/python3.6/site-packages/gensim/similarities/__init__.py:15: UserWarning: The gensim.similarities.levenshtein submodule is disabled, because the optional Levenshtein package <https://pypi.org/project/python-Levenshtein/> is unavailable. Install Levenhstein (e.g. `pip install python-Levenshtein`) to suppress this warning.
  warnings.warn(msg)
Traceback (most recent call last):
  File "predict.py", line 10, in <module>
    experiment = Experiment(Mode.PREDICT, args.config(mode=Mode.PREDICT))
  File "/home/yyoo/src/claf/claf/learn/experiment.py", line 48, in __init__
    self.load_setting()
  File "/home/yyoo/src/claf/claf/learn/experiment.py", line 89, in load_setting
    cuda_devices, checkpoint_path, prev_cuda_device_id=prev_cuda_device_id
  File "/home/yyoo/src/claf/claf/learn/experiment.py", line 101, in _read_checkpoint
    f"cuda:{prev_cuda_device_id}": f"cuda:{cuda_devices[0]}"