Logo
Explore Help
Register Sign In
mirrors/pytorch-image-models
1
0
Fork 0
You've already forked pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity
pytorch-image-models/timm/utils
History
Ross Wightman deb9895600 Update checkpoint save to fix old hard-link + fuse issue I ran into again... fix #340
2025-01-08 15:36:58 -08:00
..
__init__.py
…
agc.py
Initial AGC impl. Still testing.
2021-02-15 23:22:44 -08:00
attention_extract.py
Add regex matching support to AttentionExtract. Add return_dict support to graph extractors and use returned output in AttentionExtractor
2024-05-22 14:33:39 -07:00
checkpoint_saver.py
Update checkpoint save to fix old hard-link + fuse issue I ran into again... fix #340
2025-01-08 15:36:58 -08:00
clip_grad.py
Initial AGC impl. Still testing.
2021-02-15 23:22:44 -08:00
cuda.py
Cleanup some amp related behaviour to better support different (non-cuda) devices
2024-10-18 13:54:16 -07:00
decay_batch.py
…
distributed.py
add npu support
2024-10-16 07:13:45 +00:00
jit.py
disable nvfuser for jit te/legacy modes (for PT 1.12+)
2022-07-13 10:34:34 -07:00
log.py
Reorg of utils into separate modules
2020-09-07 13:58:09 -07:00
metrics.py
Tweak accuracy topk safety. Fix #807
2021-08-19 14:18:53 -07:00
misc.py
Davit update formatting and fix grad checkpointing (#7)
2023-01-15 14:34:56 -08:00
model_ema.py
Fix #2127 move to ema device
2024-04-10 21:29:09 -07:00
model.py
Tweak a few docstrings
2024-11-13 10:12:31 -08:00
onnx.py
Add support for dynamo based onnx export
2024-03-13 12:05:26 -07:00
random.py
Remove commented code, add more consistent seed fn
2021-04-12 09:51:36 -07:00
summary.py
…
Powered by Gitea Version: 1.23.8 Page: 1876ms Template: 42ms
English
Bahasa Indonesia Deutsch English Español Français Gaeilge Italiano Latviešu Magyar nyelv Nederlands Polski Português de Portugal Português do Brasil Suomi Svenska Türkçe Čeština Ελληνικά Български Русский Українська فارسی മലയാളം 日本語 简体中文 繁體中文(台灣) 繁體中文(香港) 한국어
Licenses API