Add longer training on first iteration

[braceal]

Expose task level parameter: initial_epochs default to the same is epochs

The logic can be added to the task level. It looks something like:

if cfg.stage_idx == 0:
    epochs = cfg.initial_epochs
else:
    epochs = cfg.epochs

Then pass epochs to the training function.

[braceal]

Done.