search

ankane / torch.rb

Deep learning for Ruby, powered by LibTorch
Other
704 stars 30 forks source link

TensorBoard #53

Open pointlessone opened 4 months ago

pointlessone commented 4 months ago

Is there a way to integrate with TwnsorBoard or something similar?

ankane commented 4 months ago

Hi @pointlessone, it's not currently supported, but happy to accept a PR (porting the PyTorch code), or I'll add it when I have some time.

https://pytorch.org/tutorials/recipes/recipes/tensorboard_with_pytorch.html

Edit: It looks like PyTorch it uses the tensorflow package for the Event protocol buffers, so it may not be trivial