Ross Wightman 2b251fb291 Wrap torch checkpoint() fn to default use_reentrant flag to False and allow env var override

2025-01-06 11:28:39 -08:00

..

data

Make sure trust_remote code only passed to HF datasets. Improve some docstrings.

2024-12-06 11:40:04 -08:00

layers

Wrap torch checkpoint() fn to default use_reentrant flag to False and allow env var override

2025-01-06 11:28:39 -08:00

loss

…

models

Wrap torch checkpoint() fn to default use_reentrant flag to False and allow env var override

2025-01-06 11:28:39 -08:00

optim

Add caution to Adan. Add decouple decay option to LAMB.

2024-12-05 13:50:30 -08:00

scheduler

…

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

py.typed

…

version.py

Back to dev version

2024-12-06 10:44:41 -08:00