ttslearn.train_util.moving_average_¶

ttslearn.train_util.moving_average_(model, model_test, beta=0.9999)[source]¶

Exponential moving average (EMA) of model parameters.

Parameters

model (torch.nn.Module) – Model to perform EMA on.
model_test (torch.nn.Module) – Model to use for the test phase.
beta (float, optional) – [description]. Defaults to 0.9999.