diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-09-17 15:59:27 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-09-17 15:59:27 +0100 |
| commit | 06cc329e715cbb820343f9849a4a45c818cb8c5e (patch) | |
| tree | bbfd12ef3467c88aa85847496a3d305f408eba20 /candle-nn/src | |
| parent | 5f83c13f17a7b16955c9b649424aca276d5e930d (diff) | |
| download | candle-06cc329e715cbb820343f9849a4a45c818cb8c5e.tar.gz candle-06cc329e715cbb820343f9849a4a45c818cb8c5e.tar.bz2 candle-06cc329e715cbb820343f9849a4a45c818cb8c5e.zip | |
Remove the parameters for the Wuerstchen layer-norm. (#879)
* Remove the parameters for the Wuerstchen layer-norm.
* Fixes.
* More fixes (including conv-transpose2d.
* More fixes.
* Again more fixes.
Diffstat (limited to 'candle-nn/src')
| -rw-r--r-- | candle-nn/src/conv.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-nn/src/conv.rs b/candle-nn/src/conv.rs index 309a5f37..31bf9af0 100644 --- a/candle-nn/src/conv.rs +++ b/candle-nn/src/conv.rs @@ -302,7 +302,7 @@ pub fn conv_transpose2d_no_bias( up: bound, }; let ws = vb.get_with_hints( - (out_channels, in_channels, kernel_size, kernel_size), + (in_channels, out_channels, kernel_size, kernel_size), "weight", init, )?; |