Hi,
I did not reproduce with default classification Jupyter notebook inside 2.0_dp docker.
After I changes
input_image_size: “3,224,224”
to
input_image_size: “3,512,512”
It can run training. Could you double check or try the classficaton notebook too?