James Betker 645d0ca767 ResidualGen mods ... - Add filters_mid spec which allows a expansion->squeeze for the transformation layers. - Add scale and bias AFTER the switch - Remove identity transform (models were converging on this) - Move attention image generation and temperature setting into new function which gets called every step with a save path		2020-06-17 17:18:28 -06:00
..
.idea
data	Add doResizeLoss to dataset	2020-06-08 11:27:06 -06:00
data_scripts	Misc	2020-06-05 21:01:50 -06:00
metrics
models	ResidualGen mods	2020-06-17 17:18:28 -06:00
options
scripts
temp
utils	Add GPU mem tracing module	2020-06-14 11:02:54 -06:00
distill_torchscript.py	Misc	2020-06-14 11:03:02 -06:00
onnx_inference.py	Misc	2020-06-14 11:03:02 -06:00
process_video.py	Debug process_video	2020-05-29 20:44:50 -06:00
requirements.txt	Introduce (untested) colab mode	2020-06-01 15:09:52 -06:00
run_scripts.sh
test.py	Misc	2020-06-02 09:35:52 -06:00
train.py	Output histograms with SwitchedResidualGenerator	2020-06-16 15:54:37 -06:00