Ross Wightman f5ca4141f7 Adjust arg order for recent vit model args, add a few comments

2022-05-02 22:41:38 -07:00

..

data

…

loss

…

models

Adjust arg order for recent vit model args, add a few comments

2022-05-02 22:41:38 -07:00

optim

…

scheduler

Remove redundant noise attr from Plateau scheduler (use parent)

2022-03-21 22:00:03 -07:00

utils

fixing channels_last on cond_conv2d; update nvfuser debug env variable

2022-04-25 12:41:46 -07:00

__init__.py

…

version.py

version 0.6.1 for master

2022-03-21 22:03:13 -07:00