Skip to content

GitLab

"docs/vscode:/vscode.git/clone" did not exist on "354775bc5755c4a6c47e008d28f27f8ccdcf8f8f"
Switch branch/tag
  1. 20 Jul, 2023 1 commit
    • ranchlai's avatar
      Add multi-label text classification support to pytorch example (#24770) · 8fd8c8e4
      ranchlai authored 
      * Add text classification example

* set the problem type and finetuning task

* ruff reformated

* fix bug for unseting label_to_id for regression

* update README.md

* fixed finetuning task

* update comment

* check if label exists in feature before removing

* add useful logging
      8fd8c8e4
  3. 18 Jul, 2023 1 commit
  5. 17 Jul, 2023 2 commits
  7. 12 Jul, 2023 1 commit
  9. 03 Jul, 2023 1 commit
  11. 30 Jun, 2023 1 commit
  13. 27 Jun, 2023 1 commit
  15. 23 Jun, 2023 1 commit
    • Matt's avatar
      Improved keras imports (#24448) · 8e164c54
      Matt authored 
      * An end to accursed version-specific imports

* No more K.is_keras_tensor() either

* Update dependency tables

* Use a cleaner call context function getter

* Add a cap to <2.14

* Add cap to examples requirements too
      8e164c54
  17. 14 Jun, 2023 1 commit
    • Patrick von Platen's avatar
      Add MMS CTC Fine-Tuning (#24281) · 1609a436
      Patrick von Platen authored 
      * Add mms ctc fine tuning

* make style

* More fixes that are needed

* make fix-copies

* make draft for README

* add new file

* move to new file

* make style

* make style

* add quick test

* make style

* make style
      1609a436
  19. 12 Jun, 2023 1 commit
  21. 07 Jun, 2023 1 commit
  23. 06 Jun, 2023 2 commits
  25. 02 Jun, 2023 2 commits
  27. 31 May, 2023 2 commits
  29. 26 May, 2023 1 commit
  31. 25 May, 2023 3 commits
  33. 24 May, 2023 1 commit
  35. 23 May, 2023 4 commits
  37. 22 May, 2023 2 commits
  39. 18 May, 2023 1 commit
  41. 16 May, 2023 1 commit
  43. 11 May, 2023 2 commits
  45. 10 May, 2023 1 commit
  47. 09 May, 2023 3 commits
  49. 08 May, 2023 1 commit
  51. 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
  53. 03 May, 2023 1 commit