Ross Wightman d4dde48dd5 Missed first_conv from resnet18d

2024-10-31 19:29:53 -07:00

..

data

Allow non train mode for wds reader to operate w/o sample count, exhaust iterator

2024-10-31 16:39:35 -07:00

layers

Cleanup some amp related behaviour to better support different (non-cuda) devices

2024-10-18 13:54:16 -07:00

loss

…

models

Missed first_conv from resnet18d

2024-10-31 19:29:53 -07:00

optim

…

scheduler

…

utils

Merge branch 'MengqingCao-npu_support' into device_amp_cleanup

2024-10-18 14:50:44 -07:00

__init__.py

…

version.py

Update version.py

2024-10-19 09:56:17 -07:00