1. 31 May, 2023 2 commits
  2. 25 May, 2023 1 commit
  3. 24 May, 2023 1 commit
  4. 22 May, 2023 1 commit
  5. 18 May, 2023 1 commit
  6. 16 May, 2023 1 commit
  7. 11 May, 2023 2 commits
  8. 10 May, 2023 1 commit
  9. 09 May, 2023 2 commits
  10. 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
  11. 03 May, 2023 1 commit
  12. 02 May, 2023 1 commit
  13. 13 Apr, 2023 1 commit
  14. 11 Apr, 2023 1 commit
  15. 05 Apr, 2023 1 commit
  16. 04 Apr, 2023 1 commit
  17. 29 Mar, 2023 1 commit
  18. 23 Mar, 2023 1 commit
  19. 22 Mar, 2023 3 commits
  20. 14 Mar, 2023 1 commit
  21. 08 Mar, 2023 1 commit
  22. 27 Feb, 2023 1 commit
  23. 24 Feb, 2023 1 commit
  24. 22 Feb, 2023 2 commits
  25. 16 Feb, 2023 1 commit
  26. 13 Feb, 2023 1 commit
  27. 10 Feb, 2023 1 commit
  28. 09 Feb, 2023 1 commit
  29. 08 Feb, 2023 1 commit
  30. 07 Feb, 2023 1 commit
  31. 06 Feb, 2023 2 commits
  32. 31 Jan, 2023 1 commit
  33. 30 Jan, 2023 1 commit