Ross Wightman 7bc7798d0e Type annotation correctness for create_act

2024-02-10 14:57:58 -08:00

..

data

Small tweak of timm ToTensor for clarity

2024-02-10 14:57:40 -08:00

layers

Type annotation correctness for create_act

2024-02-10 14:57:58 -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

Fix distributed flag bug w/ flex device handling

2024-02-03 16:26:15 -08:00

__init__.py

…

version.py

Update version.py

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