update train_csn.py: update comments

livecellx/model_zoo/segmentation/train_csn.py

@@ -156,7 +156,7 @@ def df2dataset(df):
     if args.debug:
         train_df = train_df[:100]
         val_df = val_df[:10]
-    # Get rid of last batch if it's not full to prevent BatchNorm complaints
+    # Get rid of last batch if it's not full to avoid BatchNorm complaints
     train_df = train_df.iloc[: (len(train_df) // args.batch_size) * args.batch_size]
     val_df = val_df.iloc[: (len(val_df) // args.batch_size) * args.batch_size]