The script 3_fit_simplified_posnet.py would stop if the condition if rank == 0 is added.
https://github.com/SINGROUP/StyleTranslationAugmentation/blob/28fcc0cfa82310d38f8c0095a627793c3f5d503e/3_fit_simple_posnet.py#L339
Since method checkpointer.next_epoch() also check rank, I removed the condition and let it run here. But why it stops should be understand.
The script
3_fit_simplified_posnet.pywould stop if the conditionif rank == 0is added.https://github.com/SINGROUP/StyleTranslationAugmentation/blob/28fcc0cfa82310d38f8c0095a627793c3f5d503e/3_fit_simple_posnet.py#L339
Since method
checkpointer.next_epoch()also check rank, I removed the condition and let it run here. But why it stops should be understand.