Ross Wightman e748805be3 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
..
__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	…
model_ema.py	Fix #2127 move to ema device	2024-04-10 21:29:09 -07:00
onnx.py	Add support for dynamo based onnx export	2024-03-13 12:05:26 -07:00
random.py	…
summary.py	…