fzyzcjy 8880a5cd5c Update scheduler.py		2024-03-23 11:27:33 +08:00
..
data	Small tweak of timm ToTensor for clarity	2024-02-10 14:57:40 -08:00
layers	Make NormMlpClassifier head reset args consistent with ClassifierHead	2024-02-10 16:25:33 -08:00
loss	Add pos_weight and support for summing over classes to BCE impl in train scripts	2023-12-30 12:13:06 -08:00
models	Improve vit conversions. OpenAI convert pass through main convert for patch & pos resize. Fix #2120	2024-03-21 10:00:43 -07:00
optim	…
scheduler	Update scheduler.py	2024-03-23 11:27:33 +08:00
utils	Add support for dynamo based onnx export	2024-03-13 12:05:26 -07:00
__init__.py	…
version.py	Update version.py	2024-02-19 11:08:17 -08:00