Ross Wightman d0ff315eed Merge remote-tracking branch 'emav3/faster_ema' into mesa_ema		2024-01-27 14:52:10 -08:00
..
data	Pass train-crop-mode to create_loader/transforms from train.py args	2024-01-24 16:19:02 -08:00
layers	Fix meshgrid deprecation warnings and backward compat with explicit 'ndgrid' and 'meshgrid' fn w/o indexing arg	2024-01-27 13:48:33 -08:00
loss	Add pos_weight and support for summing over classes to BCE impl in train scripts	2023-12-30 12:13:06 -08:00
models	Fix meshgrid deprecation warnings and backward compat with explicit 'ndgrid' and 'meshgrid' fn w/o indexing arg	2024-01-27 13:48:33 -08:00
optim	Update sgdw for older pytorch	2023-12-11 12:10:29 -08:00
scheduler	Allow training w/o validation split set	2024-01-08 09:38:42 -08:00
utils	Merge remote-tracking branch 'emav3/faster_ema' into mesa_ema	2024-01-27 14:52:10 -08:00
__init__.py	…
version.py	Update version.py	2024-01-08 09:39:22 -08:00