Tao unet "TrainingConfig" has no field named "activation"

Yes, thanks for the time to narrow down.

Please refer to
Problems encountered in training unet and inference unet - #27 by Morganh .

I cannot reproduce the accuracy drop when exporting. Also, I was training a model of 960x544.

For sigmoid activation, I will check if I can reproduce later.