Ross Wightman b5a4fa9c3b Add pos_weight and support for summing over classes to BCE impl in train scripts

2023-12-30 12:13:06 -08:00

..

data

Add in_chans to data config helper. Fix #2021

2023-11-23 12:44:59 -08:00

layers

Add device arg for patch embed resize, fix #2024

2023-12-04 11:42:13 -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

Add parsable json results output for train.py, tweak --pretrained-path to force head adaptation

2023-12-22 11:18:25 -08:00

optim

Update sgdw for older pytorch

2023-12-11 12:10:29 -08:00

scheduler

Scheduler update, add v2 factory method, support scheduling on updates instead of just epochs. Add LR to summary csv. Add lr_base scaling calculations to train script. Fix #1168

2022-10-07 10:43:04 -07:00

utils

…

__init__.py

Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models

2022-12-06 15:00:06 -08:00

version.py

Version 0.9.13dev0

2023-11-25 10:52:31 -08:00