Ross Wightman 0cbf4fa586 _orig_mod still causing issues even though I thought it was fixed in pytorch, add unwrap / clean helpers		2024-07-19 11:03:45 -07:00
..
__init__.py	Add AttentionExtract helper module	2024-05-04 14:10:00 -07:00
agc.py	…
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	…
clip_grad.py	…
cuda.py	…
decay_batch.py	…
distributed.py	Tweak dist_backend to use device_type (before possible :)	2024-05-15 08:49:25 -07:00
jit.py	…
log.py	…
metrics.py	…
misc.py	…
model.py	_orig_mod still causing issues even though I thought it was fixed in pytorch, add unwrap / clean helpers	2024-07-19 11:03:45 -07:00
model_ema.py	…
onnx.py	…
random.py	…
summary.py	…