Ross Wightman c16e965037 Add some ViT comments and fix a few minor issues.

2021-01-24 23:18:35 -08:00

..

data

Convert samples/targets in ParserImageInTar to numpy arrays, slightly less mem usage for massive datasets. Add a few more se/eca model defs to resnet.py

2021-01-22 16:54:33 -08:00

loss

…

models

Add some ViT comments and fix a few minor issues.

2021-01-24 23:18:35 -08:00

optim

Configure create_optimizer with args.opt_args

2020-12-08 00:03:09 -06:00

scheduler

…

utils

…

__init__.py

…

version.py

Update version to 0.3.3

2021-01-03 12:43:44 -08:00