Ross Wightman 7cf683628f Cautious optimizer impl plus some typing cleanup.

2024-11-28 15:11:58 -08:00

..

data

Add mini imagenet info files

2024-11-25 10:53:28 -08:00

layers

…

loss

…

models

Update torchvision resnet legacy weight urls in resnet.py

2024-11-26 15:53:54 +01:00

optim

Cautious optimizer impl plus some typing cleanup.

2024-11-28 15:11:58 -08:00

scheduler

Extend train epoch schedule by warmup_epochs if warmup_prefix enable, allows schedule to reach end w/ prefix enabledy

2024-11-08 11:01:11 -08:00

utils

Tweak a few docstrings

2024-11-13 10:12:31 -08:00

__init__.py

…

py.typed

Add py.typed file as recommended by PEP 561

2024-11-14 11:26:00 -08:00

version.py

…