Hi all,
I had the time to go ahead and continue my customisation for this project, once I set the task in the conf into ((ntasks: 1) the program completes successfully the training and validation. However, the learning rate is still very low as showing above,
If I increase the learning rate (ntasks: 5) I am getting the below error on training task1:

in nll_loss
ret = torch._C._nn.nll_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index)
IndexError: Target 17 is out of bounds.

Thank you all in advance.