Ross Wightman 211cf90721 Imports getting unwieldy in vision_transformer.py

2025-05-30 15:11:51 -07:00

..

data

remove download argument from torch_kwargs

2025-05-09 07:34:32 -07:00

layers

Add EVA ViT based PE (Perceptual Encoder) impl

2025-05-10 13:13:12 -07:00

loss

Fix #2132, remove use of _C.set_grad_enable. Line endings were messed up too

2024-04-09 09:00:23 -07:00

models

Imports getting unwieldy in vision_transformer.py

2025-05-30 15:11:51 -07:00

optim

Fix issue taking device from V before V exists

2025-01-31 12:52:47 -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

Remove torch_out from onnx export, no point without the export_ fn

2025-04-15 14:01:17 -07:00

__init__.py

Use import alias

2025-01-22 10:27:17 -08:00

py.typed

Add py.typed file as recommended by PEP 561

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

version.py

Update version.py

2025-02-22 10:50:21 -08:00