TAO hangs at end of training

There is no update from you for a period, assuming this is not an issue any more.
Hence we are closing this topic. If need further support, please open a new one.
Thanks

Sorry, I am a little confused. If you train for 100 epochs, how about its result? Does 1st epoch, 2nd epoch , 98th epoch work well?