Ross Wightman 015fbe457a Merge branch 'MengqingCao-npu_support' into device_amp_cleanup		2024-10-18 14:50:44 -07:00
..
__init__.py	…
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	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	…
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	…