You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
* Support for native Torch AMP and channels_last memory format added to train/validate scripts
9
+
* Support for native Torch AMP and channels_last memory format added to train/validate scripts (`--channels-last`, ``--native-amp` vs `--apex-amp`)
10
10
* Models tested with channels_last on latest NGC 20.08 container. AdaptiveAvgPool in attn layers changed to mean((2,3)) to work around bug with NHWC kernel.
0 commit comments