LaurentMazare / ocaml-torch

OCaml bindings for PyTorch
Apache License 2.0
412 stars 38 forks source link

RL examples no longer compile (again) #55

Open kyliedunn526 opened 3 years ago

kyliedunn526 commented 3 years ago

Eg. running ocaml-torch/_build/default/examples/reinforcement-learning/dqn_atari.exe produces

actions: NOOP,FIRE,RIGHT,LEFT,RIGHTFIRE,LEFTFIRE 0 0 (0/0 frames) zsh: segmentation fault

EDIT: Tried with char_rnn and it works fine, so possibly unique to RL?

thuangb commented 2 years ago

memory issues, try better GPU and CPU