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] [2]

Edited 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