tch-rs

tch-rs copied to clipboard

Broadcasting seems to mess with gradients

1

Hello, I'm not entirely sure this is a bug or if this is me just misusing the API. I am training a pretty simple full connected layer with some data...

ZagButNoZig

Hello, I believe it would be helpful for who is depending on this library to keep a changelog of what is added, removed, changed and fixed. This is a possible...

vcarpani

Add more function from torch::cuda

3

This PR adds some more torch::cuda functions and improves documentation for the existing ones.

vcarpani

Hi, I want to load a JIT model with tch-rs but get the following error: ``` thread 'main' panicked at 'cannot load model: Torch("\nUnknown builtin op: torch_scatter::scatter_max.\nCould not find any...

KeKsBoTer

I'm trying to implement a3c. There's one Global VarStore and each agent's local VarStores. My implementation is like > 1. agent action 2. get rewards 3. calc loss and grad...

HyeokSuLee

I think this function should also needs to return error when target is empty. ```rust //make src varstore let vs = nn::VarStore::new(device); // src variables created let _ = model(&vs.root(),...

HyeokSuLee

using bevy with tch-rs. Use bevy for a3c. struct `GlobalParam` for global weights. In thread function, collect actions with agent and send them to global. Then global update weights with...

HyeokSuLee

Remove the dummy dependency hack and use extra-ling-args instead.

2

This is only available in cargo 1.50 and in nightly mode at the moment. This can be tried out via: ```bash cargo +nightly run --example basics -Zextra-link-arg --verbose ```

LaurentMazare

Compilation error

8

I'm running macOS v12.3 on an M1 macBook. M1-compatible version of libtorch is installed via homebrew. I'm running a basic example via `cargo run --example basics`, and get the following...

ghost

Documentation for use on Apple Silicon

26

I'm doing some development on an M1 Mac. And, I'm having some trouble installing `LibTorch`. Though I assumed it wouldn't work, I tried just installing from the link that usually...

l-monninger