Ross Wightman
|
7a9c6811c9
|
Add eps arg to LayerNorm2d, add 'tf' (tensorflow) variant of trunc_normal_ that applies scale/shift after sampling (instead of needing to move a/b)
|
2022-07-01 15:15:39 -07:00 |
|
Ross Wightman
|
cf5fec5047
|
Cleanup experimental vit weight init a bit
|
2021-03-20 09:44:24 -07:00 |
|
Ross Wightman
|
17270c69b9
|
Remove annoying InceptionV3 dependency on scipy and insanely slow trunc_norm init. Bring InceptionV3 code into this codebase and use upcoming torch trunch_norm_ init.
|
2020-05-12 21:59:34 -07:00 |