Ross Wightman dd84ef2cd5 ModelEmaV3 and MESA experiments

2024-02-02 09:45:04 -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

ModelEmaV3 and MESA experiments

2024-02-02 09:45:04 -08:00

__init__.py

…

version.py

Update version.py

2024-01-08 09:39:22 -08:00