1. 09 May, 2023 2 commits
  2. 05 May, 2023 1 commit
    • Ashwin Mathur's avatar
      Add `no_trainer` scripts to pre-train Vision Transformers (#23156) · fc6c8b0e
      Ashwin Mathur authored
      
      
      * Add run_mim_no_trainer.py draft from #20412
      
      Add parse_args method and copy over other dependencies
      
      Add Method call for sending telemetry
      
      Initialize Accelerator
      
      Make one log on every process
      
      Set seed and Handle repository creation
      
      Initialize dataset and Set validation split
      
      Create Config
      
      Adapt Config
      
      Update Config
      
      Create Feature Extractor
      
      Create model
      
      Set column names
      
      Create transforms
      
      Create mask generator
      
      Create method to preprocess images
      
      Shuffle datasets if needed and set transforms
      
      Create Dataloaders
      
      Add optimizer
      
      Add learning rate scheduler
      
      Prepare everything with our accelerator
      
      Tie weights for TPU training
      
      Recalculate training steps and training epochs
      
      Set accelerator checkpointing steps
      
      Initialize trackers and store configuration
      
      Set total batch size
      
      Fix typo: mlm -> mim
      
      Log info at the start of training
      
      Load in the weights and states from previous save
      
      update the progress_bar if load from checkpoint
      
      Define train loop
      
      Add evaluation loop to training
      
      Add to parse_args method
      
      Push repo to hub
      
      Save accelerator state
      
      End training and save model and feature extractor
      
      Remove unused imports
      
      Fix trailing whitespace
      
      * Update code based on comments, Rename feature_extractor to image_processor
      
      * Fix linting
      
      * Add argument for learning rate
      
      * Add argument for setting number of training epochs
      
      * Remove incorrect logger argument
      
      * Convert max_train_steps to int for tqdm
      
      ---------
      Co-authored-by: default avatarSaad Mahmud <shuvro.mahmud79@gmail.com>
      fc6c8b0e
  3. 03 May, 2023 1 commit
  4. 02 May, 2023 1 commit
  5. 13 Apr, 2023 1 commit
  6. 11 Apr, 2023 1 commit
  7. 05 Apr, 2023 1 commit
  8. 04 Apr, 2023 1 commit
  9. 29 Mar, 2023 1 commit
  10. 23 Mar, 2023 1 commit
  11. 22 Mar, 2023 3 commits
  12. 14 Mar, 2023 1 commit
  13. 08 Mar, 2023 1 commit
  14. 27 Feb, 2023 1 commit
  15. 24 Feb, 2023 1 commit
  16. 22 Feb, 2023 2 commits
  17. 16 Feb, 2023 1 commit
  18. 13 Feb, 2023 1 commit
  19. 10 Feb, 2023 1 commit
  20. 09 Feb, 2023 1 commit
  21. 08 Feb, 2023 1 commit
  22. 07 Feb, 2023 1 commit
  23. 06 Feb, 2023 2 commits
  24. 31 Jan, 2023 1 commit
  25. 30 Jan, 2023 1 commit
  26. 23 Jan, 2023 2 commits
  27. 19 Jan, 2023 1 commit
  28. 18 Jan, 2023 1 commit
  29. 06 Jan, 2023 1 commit
  30. 05 Jan, 2023 2 commits
  31. 03 Jan, 2023 1 commit
  32. 24 Dec, 2022 1 commit
  33. 21 Dec, 2022 1 commit