tch-rs icon indicating copy to clipboard operation
tch-rs copied to clipboard

Published 20 hours ago • verified publisher icon LaurentMazare

Added missing InstanceNorm module.

Open joverwey opened this issue 3 years ago • 1 comments

I noticed that InstanceNorm was not implemented. I tried to follow the same pattern as BatchNorm with a few exceptions. For BatchNorm, the current implementation always uses weights and biases and always keeps track of running stats. In the PyTorch implementation this is optional. For InstanceNorm I do need to be able to enable/disable this so I added flags for

  • affine
  • track_running_stats

like the PyTorch implementation has.

Let me know if I missed anything.

joverwey avatar Aug 19 '22 03:08 joverwey

Clippy seems to be failing in a file I didn't touch? Any idea how I can fix this?

joverwey avatar Oct 12 '22 23:10 joverwey