In tensorflow 2.4.0, detected Infinity or NaN in output of tf.math.multiply_no_nan?

Question

def basic_block(x, output_channel, is_training=True, stride=1, downsample=None):
    residual = x
    out = cbr_layer(x, output_channel, 3, stride=stride, is_trainin