Issue in transfer learning command 📈 Enable Model EMA

[dsbyprateekg]

Hi,

I am trying the transfer learning command for my custom dataset training using my dataset config file (lego.yaml) in the location yolo/config/dataset/ -

!python yolo/lazy.py task=train task.data.batch_size=8 model=v9-c dataset=lego device=cuda

But the above command generate following one-liner in the output log file- :chart_with_upwards_trend: Enable Model EMA

Can you please help me to resolve this issue?

[henrytsui000]

Hi,

That is expected behavior. EMA (Exponential Moving Average) is a strategy designed to help validate model performance from earlier epochs and provides a more balanced representation of the model's parameters. This strategy typically has a positive effect on training and evaluation and won't effect origin weight.

If you prefer not to use EMA, you can disable it with the following command:

[other command] task.ema.enable=False

Best regards,
Henry Tsui

[dsbyprateekg]

Hi,

That is expected behavior. EMA (Exponential Moving Average) is a strategy designed to help validate model performance from earlier epochs and provides a more balanced representation of the model's parameters. This strategy typically has a positive effect on training and evaluation and won't effect origin weight.

If you prefer not to use EMA, you can disable it with the following command:

[other command] task.ema.enable=False

Best regards, Henry Tsui

my issue was, that training is not started.
And it is showing only this line in the console- 📈 Enable Model EMA