Hi Yuntao,
I crop the tomogram to a smaller size,
/vast/palmer/apps/avx2/software/TensorFlow/2.5.0-fosscuda-2020b/lib/python3.8/site-packages/tensorflow/python/keras/utils/generic_utils.py:494: CustomMaskWarning: Custom mask layers require a config and must override get_config. When loading, the custom mask layer must be passed to the custom_objects argument.
warnings.warn('Custom mask layers require a config and must override '
The error only occurs at the last epoch of iteration 1, does this indicate a mask problem?