Ross Wightman 88889de923 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
..
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	Allow training w/o validation split set	2024-01-08 09:38:42 -08:00
__init__.py	…
version.py	Update version.py	2024-01-08 09:39:22 -08:00