train.py force dataset-resampled

Author: afang-story

train.py

@@ -177,12 +177,6 @@ def save_training_artifacts(args, config, checkpoint):
         help="Number of times we save checkpoints during training.",
     )
     parser.add_argument("--seed", type=int, default=0, help="Random seed.")
-    parser.add_argument(
-        "--dataset_resampled",
-        default=False,
-        action="store_true",
-        help="Whether to use sampling with replacement for webdataset shard selection.",
-    )
     parser.add_argument(
         "--report_to_wandb",
         default=False,
@@ -285,8 +279,7 @@ def save_training_artifacts(args, config, checkpoint):
         "--resume",
         f"{args.resume}",
     ]
-    if args.dataset_resampled:
-        main_args.append("--dataset-resampled")
+    main_args.append("--dataset-resampled")
     if args.report_to_wandb:
         main_args.extend(
             [