1. 28 Jul, 2023 2 commits
  2. 20 Jul, 2023 1 commit
  3. 17 Jul, 2023 1 commit
  4. 12 Jun, 2023 1 commit
  5. 07 Jun, 2023 1 commit
  6. 06 Jun, 2023 1 commit
  7. 22 May, 2023 1 commit
  8. 09 May, 2023 1 commit
  9. 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
  10. 13 Apr, 2023 1 commit
  11. 14 Mar, 2023 1 commit
  12. 22 Feb, 2023 2 commits
  13. 10 Feb, 2023 1 commit
  14. 23 Jan, 2023 1 commit
  15. 19 Jan, 2023 1 commit
  16. 01 Dec, 2022 1 commit
  17. 02 Nov, 2022 1 commit
    • amyeroberts's avatar
      Add Image Processors (#19796) · a6b77598
      amyeroberts authored
      
      
      * Add CLIP image processor
      
      * Crop size as dict too
      
      * Update warning
      
      * Actually use logger this time
      
      * Normalize doesn't change dtype of input
      
      * Add perceiver image processor
      
      * Tidy up
      
      * Add DPT image processor
      
      * Add Vilt image processor
      
      * Tidy up
      
      * Add poolformer image processor
      
      * Tidy up
      
      * Add LayoutLM v2 and v3 imsge processors
      
      * Tidy up
      
      * Add Flava image processor
      
      * Tidy up
      
      * Add deit image processor
      
      * Tidy up
      
      * Add ConvNext image processor
      
      * Tidy up
      
      * Add levit image processor
      
      * Add segformer image processor
      
      * Add in post processing
      
      * Fix up
      
      * Add ImageGPT image processor
      
      * Fixup
      
      * Add mobilevit image processor
      
      * Tidy up
      
      * Add postprocessing
      
      * Fixup
      
      * Add VideoMAE image processor
      
      * Tidy up
      
      * Add ImageGPT image processor
      
      * Fixup
      
      * Add ViT image processor
      
      * Tidy up
      
      * Add beit image processor
      
      * Add mobilevit image processor
      
      * Tidy up
      
      * Add postprocessing
      
      * Fixup
      
      * Fix up
      
      * Fix flava and remove tree module
      
      * Fix image classification pipeline failing tests
      
      * Update feature extractor in trainer scripts
      
      * Update pad_if_smaller to accept tuple and int size
      
      * Update for image segmentation pipeline
      
      * Update src/transformers/models/perceiver/image_processing_perceiver.py
      Co-authored-by: default avatarAlara Dirik <8944735+alaradirik@users.noreply.github.com>
      
      * Update src/transformers/image_processing_utils.py
      Co-authored-by: default avatarNielsRogge <48327001+NielsRogge@users.noreply.github.com>
      
      * Update src/transformers/models/beit/image_processing_beit.py
      Co-authored-by: default avatarNielsRogge <48327001+NielsRogge@users.noreply.github.com>
      
      * PR comments - docstrings; remove accidentally added resize; var names
      
      * Update docstrings
      
      * Add exception if size is not in the right format
      
      * Fix exception check
      
      * Fix up
      
      * Use shortest_edge in tuple in script
      Co-authored-by: default avatarAlara Dirik <8944735+alaradirik@users.noreply.github.com>
      Co-authored-by: default avatarNielsRogge <48327001+NielsRogge@users.noreply.github.com>
      a6b77598
  18. 01 Nov, 2022 1 commit
  19. 10 Oct, 2022 1 commit
  20. 14 Sep, 2022 1 commit
  21. 06 Aug, 2022 1 commit
  22. 27 Jul, 2022 1 commit
  23. 16 Jun, 2022 1 commit
  24. 07 Jun, 2022 1 commit
    • Sylvain Gugger's avatar
      Add examples telemetry (#17552) · 3cab9027
      Sylvain Gugger authored
      * Add examples telemetry
      
      * Alternative approach
      
      * Add to all other examples
      
      * Add to templates as well
      
      * Put framework separately
      
      * Same for TensorFlow
      3cab9027
  25. 12 May, 2022 2 commits
  26. 06 Apr, 2022 1 commit
  27. 04 Apr, 2022 1 commit
  28. 03 Mar, 2022 1 commit
  29. 17 Feb, 2022 1 commit
    • NielsRogge's avatar
      Add SimMIM (#15586) · 57882177
      NielsRogge authored
      
      
      * Add first draft
      
      * Make model importable
      
      * Make SwinForMaskedImageModeling importable
      
      * Fix imports
      
      * Add missing inits
      
      * Add support for Swin
      
      * Fix bug
      
      * Fix bug
      
      * Fix another bug
      
      * Fix Swin MIM implementation
      
      * Fix default encoder stride
      
      * Fix Swin
      
      * Add print statements for debugging
      
      * Add image_size data argument
      
      * Fix Swin
      
      * Fix image_size
      
      * Add print statements for debugging
      
      * Fix print statement
      
      * Remove print statements
      
      * Improve reshaping of bool_masked_pos
      
      * Add support for DeiT, fix tests
      
      * Improve docstrings
      
      * Apply new black version
      
      * Improve script
      
      * Fix bug
      
      * Improve README
      
      * Apply suggestions from code review
      
      * Remove DS_Store and add to gitignore
      
      * Apply suggestions from code review + fix BEiT Flax
      
      * Revert BEiT changes
      
      * Improve README
      
      * Fix code quality
      
      * Improve README
      Co-authored-by: default avatarNiels Rogge <nielsrogge@Nielss-MBP.localdomain>
      Co-authored-by: default avatarNiels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
      57882177
  30. 27 Jan, 2022 2 commits
  31. 21 Jan, 2022 1 commit
    • NielsRogge's avatar
      [ViTMAE] Add image pretraining script (#15242) · 6c7b68d4
      NielsRogge authored
      * Add script
      
      * Improve script
      
      * Fix data collator
      
      * Update README
      
      * Add label_names argument
      
      * Apply suggestions from code review
      
      * Add config parameters
      
      * Update script
      
      * Fix bug
      
      * Improve README
      
      * Improve README and add test
      
      * Fix import
      
      * Add image_column_name
      6c7b68d4