[docs] Fix datasets in guides (#28715)

* change datasets * fix
2024-01-26 09:29:07 -08:00
parent f8b7c4345a
commit abe0289e6d
11 changed files with 112 additions and 75 deletions
--- a/examples/pytorch/image-pretraining/run_mim_no_trainer.py
+++ b/examples/pytorch/image-pretraining/run_mim_no_trainer.py
@@ -627,7 +627,9 @@ def main():
        name=args.lr_scheduler_type,
        optimizer=optimizer,
        num_warmup_steps=args.num_warmup_steps * accelerator.num_processes,
-        num_training_steps=args.max_train_steps if overrode_max_train_steps else args.max_train_steps * accelerator.num_processes,
+        num_training_steps=args.max_train_steps
+        if overrode_max_train_steps
+        else args.max_train_steps * accelerator.num_processes,
    )

    # Prepare everything with our `accelerator`.