tch-rs

tch-rs copied to clipboard

Exposed group norm layer

4

lostmsu

Supply custom gradient in the backward function

1

Recently I've been working on a project where I need to supply custom gradient into the back propagation process. Therefore I need an advanced `backward` function where I can pass...

Liby99

adamnemecek

Use windows unicode paths to get around windows 260 char path limit.

1

`libtorch` fails to extract on Windows when the project is near the 260 char windows path limit, this adds a windows check and adjusts the path. Secondly, this normalizes paths...

MarkJGx

Preliminary support for stream guard.

6

This adds some basic support for cuda streams #234. Note that this has not been tested yet and is even unlikely to compile as I don't have access to a...

LaurentMazare

- Scaling factor is initialized to 1 instead of [0, 1] uniform I discovered an issue that the running_var and scaling factor converge to very small values. [Here](https://imgur.com/yU0Zutc) is an...

jerry73204

- Use serde + envy to parse environment variables - Propagate errors to main() - Organize "cargo:*" printing

jerry73204

This PR upgrades dependencies using `cargo upgrade` from [cargo-edit](https://crates.io/crates/cargo-edit). Note that it introduces breaking changes due to `cpython` and `half` update, requiring minor version bump.

jerry73204

Revise TryFrom impls and make ndarray optional

2

`TryFrom` now receives a referenced type instead of owned type. The move is unnecessary because it always copies the data. `TryInto` impls are removed in flavor of `TryFrom`. That's because...

jerry73204

Add WGAN-GP example

1

I'd like to present the Wasserstein GAN with gradient penalty example. It is slightly modified from the [gan example](https://github.com/LaurentMazare/tch-rs/blob/master/examples/gan/main.rs). It removes batch norms in discriminator as the paper suggested, and...

jerry73204