Commit Graph

16 Commits (8412d44dced167353b914b798f86c72f7f89a058)

Author SHA1 Message Date
Glenn Jocher 41ff83a98d update torch_utils.py to FP16 EMA 5 years ago
Glenn Jocher a586751904 multi-gpu ckpt filesize bug fix 5 years ago
Glenn Jocher 24c5a941f0 --resume EMA fix 5 years ago
Glenn Jocher 121d90b3f2 update fuse_conv_and_bn() 5 years ago
Glenn Jocher 38f5c1ad1d pruning and sparsity initial commit 5 years ago
NanoCode012 ccfa9373ae
Update to remove "module" attribute check 5 years ago
NanoCode012 f83521ab9a
Fix save error for multi-gpu 5 years ago
Glenn Jocher af41083d52 EMA FP16 fix 5 years ago
Glenn Jocher df224a0d8f EMA bug fix 5 years ago
Glenn Jocher f02481c73a
Update torch_utils.py 5 years ago
Lornatang 8b8b792802 There is no need to download extra packages, official bring it with you 5 years ago
Glenn Jocher 8fe299f179 model fuse 5 years ago
Glenn Jocher b810b21299 augmented inference 5 years ago
Glenn Jocher c12023f145 updates 5 years ago
Glenn Jocher 22d6088205 speed-reproducibility fix 5 years ago
Glenn Jocher 1e84a23f38 initial commit 5 years ago