Ross Wightman 711c5dee6d Update sgdw for older pytorch

2023-12-11 12:10:29 -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

…

models

Add --pretrained-path arg to train script to allow passing local checkpoint as pretrained. Add missing/unexpected keys log.

2023-12-11 12:10:29 -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

MobileOne and FastViT weights on HF hub, more code cleanup and tweaks, features_only working. Add reparam flag to validate and benchmark, support reparm of all models with fuse(), reparameterize() or switch_to_deploy() methods on modules

2023-08-23 22:50:37 -07:00

__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