Xihua Dong 0c136f7dab fix img_size type		2025-04-18 21:05:05 -07:00
..
data	Fix typos	2025-01-19 13:39:40 -08:00
layers	fix img_size type	2025-04-18 21:05:05 -07:00
loss	…
models	Add local-dir: schema support for model loading (config + weights) from folder	2025-04-17 12:32:19 -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