Public
Authored by Mika Juuti

PyTorch: Temporal Ensembling

A PyTorch3 implementation of the temporal ensembling [1]. This snippets exists because I could not find a pytorch implementation of temporal ensembling. The authors' original code is available for tensorflow at [2].

Keeps training data in memory for efficiency.

[1] https://arxiv.org/abs/1610.02242 [2] https://github.com/smlaine2/tempens/blob/master/train.py

Edited
tempens_train.py 9.87 KB
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment