Internal change

official/vision/beta/projects/basnet/losses/basnet_losses.py

@@ -52,6 +52,7 @@ class BASNetLoss:
     total_iou_loss = tf.math.add_n(iou_losses)
     
     total_loss = total_bce_loss + total_ssim_loss + total_iou_loss
+    total_loss = total_loss / len(levels)
 
     return total_loss
     

official/vision/beta/projects/basnet/modeling/basnet_model.py

@@ -30,7 +30,7 @@ class BASNetModel(tf.keras.Model):
   def __init__(self,
                backbone,
                decoder,
-               refinement,
+               refinement=None,
                **kwargs):
     """BASNet initialization function.