ocaml-torch
ocaml-torch copied to clipboard
LaurentMazare
OCaml bindings for PyTorch
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?
Hello! Earlier I had installed ocaml-torch for a CPU version. I am trying to install ocaml-torch for GPU acceleration with the following commands. (Downloaded and unzipped libtorch 1.10) ``` export...
Hi, It seems to me that ocaml-torch does not provide API for torch.distributions. Specifically, it does not implement various probability distributions with some common functions such as sample and log_prob....
Programs I write using ocaml-torch that use GPU acceleration segfault right before terminating: ``` Segmentation fault (core dumped) ``` This is not a huge deal as it happens when the...
Run some training example on the whole ImageNet dataset. There may be some performance implications, e.g. loading images should be done in parallel.
The current tensor type is `Tensor.t`. However tensors can embed multiple kind of elements and calling a function like `to_float0_exn` on a tensor containing integers is likely to raise. We...
Hi, I installed libtorch on a Mac with ARM M1 chip using opam and found that intel binaries were actually installed. I initially posted the issue on the torch repository...
Hello, I've been trying to shift a hybrid ocaml-python program to mostly ocaml. Part of this program is a simple image collision test; when implemented it in ocaml, it is...
I'm able to opam install ocaml-torch on my intel mac, macos 12.6.3 when i first uninstall other pytorch distributions on the system - anaconda installed pytorch - homebrew installed pytorch...
pytorch 1.13 has a support for gpu acceleration on macos, including the new ARM processors. i have sucessfully installed pytorch via homebrew on an intel imac. i have shared libraries...