James Betker 1b1431133b Add DualOutputSRG ... Also removes the old multi-return mechanism that Generators support. Also fixes AttentionNorm.		2020-07-14 09:28:24 -06:00
..
__init__.py	mmsr	2019-08-23 21:42:47 +08:00
distill_torchscript.py	Misc changes	2020-07-10 09:45:48 -06:00
gpu_mem_track.py	Add GPU mem tracing module	2020-06-14 11:02:54 -06:00
numeric_stability.py	Add DualOutputSRG	2020-07-14 09:28:24 -06:00
onnx_inference.py	Misc changes	2020-07-10 09:45:48 -06:00
util.py	Introduce (untested) colab mode	2020-06-01 15:09:52 -06:00