# TrainingArgument does not work on colab **URL:** **Author:** ![Abdulbaqi](https://sea2.discourse-cdn.com/hellohellohello/user_avatar/discuss.huggingface.co/abdulbaqi/32/16866_2.png) [@Abdulbaqi](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2FAbdulbaqi)%5C **Post date:** [June 15, 2023, 12:00pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Ftrainingargument-does-not-work-on-colab%2F43372%2F1 "2023-06-15T12:00:34Z") I was following chapter 3 of the NLP course on Fine-tuning a model with the Trainer API and when executing the command in colab > training\_args = TrainingArguments(“test-trainer”) I get: > ImportError Traceback (most recent call last) > in \() > ----\> 1 training\_args = TrainingArguments(output\_dir=“test-trainer”) > > 4 frames > /usr/local/lib/python3.10/dist-packages/transformers/training\_args.py in \_setup\_devices(self) > 1670 if not is\_sagemaker\_mp\_enabled(): > 1671 if not is\_accelerate\_available(min\_version=“0.20.1”): > → 1672 raise ImportError( > 1673 “Using the `Trainer` with `PyTorch` requires `accelerate>=0.20.1`: Please run `pip install transformers[torch]` or `pip install accelerate -U`” > 1674 ) > > ImportError: Using the `Trainer` with `PyTorch` requires `accelerate>=0.20.1`: Please run `pip install transformers[torch]` or `pip install accelerate -U` > > * * * > > NOTE: If your import is failing due to a missing package, you can > manually install dependencies using either !pip or !apt. > > ## To view examples of installing some common dependencies, click the “Open Examples” button below. I try to pip install the recommended ones but never work for me. Help! ---
**Author:** ![precisiondata](https://avatars.discourse-cdn.com/v4/letter/p/ec9cab/32.png) [@precisiondata](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fprecisiondata)%5C **Post date:** [June 16, 2023, 4:31pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Ftrainingargument-does-not-work-on-colab%2F43372%2F2 "2023-06-16T16:31:44Z")
I experienced the same import error when running the following script from the hugging face transformer quick tour. Installing the 2 mentioned libraries did not resolve. from transformers import TrainingArguments training\_args = TrainingArguments( output\_dir=“./pt\_training”, learning\_rate=2e-5, per\_device\_train\_batch\_size=8, per\_device\_eval\_batch\_size=8, num\_train\_epochs=2, ) ---
**Author:** ![zcongfly](https://avatars.discourse-cdn.com/v4/letter/z/ed8c4c/32.png) [@zcongfly](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fzcongfly)%5C **Post date:** [June 18, 2023, 9:17am UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Ftrainingargument-does-not-work-on-colab%2F43372%2F3 "2023-06-18T09:17:42Z")
I encountered the same problem when I run the code below. from transformers import TrainingArguments training\_args = TrainingArguments(“test-trainer”) ERROR：Using the `Trainer` with `PyTorch` requires `accelerate>=0.20.1`: Please run `pip install transformers[torch]` or `pip install accelerate -U` But installing the 2 mentioned libraries did not resolve. I don’t know what to do now. Help! ---
**Author:** ![BajTomasz](https://avatars.discourse-cdn.com/v4/letter/b/b5a626/32.png) [@BajTomasz](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2FBajTomasz)%5C **Post date:** [June 18, 2023, 11:20am UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Ftrainingargument-does-not-work-on-colab%2F43372%2F4 "2023-06-18T11:20:16Z")
For me help reset environment and install older transformers !pip install transformers==4.17 ---
**Author:** ![precisiondata](https://avatars.discourse-cdn.com/v4/letter/p/ec9cab/32.png) [@precisiondata](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fprecisiondata)%5C **Post date:** [June 19, 2023, 1:15pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Ftrainingargument-does-not-work-on-colab%2F43372%2F5 "2023-06-19T13:15:54Z")
!pip install transformers==4.17 did not resolve the import error. I forgot to mention I am running the code in Google Colab. ---
**Author:** ![MatthewVedder](https://avatars.discourse-cdn.com/v4/letter/m/96bed5/32.png) [@MatthewVedder](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2FMatthewVedder)%5C **Post date:** [June 19, 2023, 11:26pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Ftrainingargument-does-not-work-on-colab%2F43372%2F6 "2023-06-19T23:26:59Z")
I’m getting the same error in the colab notebook for the **NLP Course: Lesson 3 - Fine-tuning model with the Trainer API**. In order to get around it you can: 1. Run `pip install accelerate -U` in a cell 2. In the top menu click **Runtime** → **Restart Runtime** 3. Do not rerun any cells with `!pip install` in them 4. Rerun all the other code cells and you should be good to go! On a side note, be sure to turn on a GPU for this notebook by clicking **Edit** → **Notebook Settings** → **GPU type** - from the top menu. This was the next thing that got me : ) ---
**Author:** ![kaiku03](https://sea2.discourse-cdn.com/hellohellohello/user_avatar/discuss.huggingface.co/kaiku03/32/52173_2.png) [@kaiku03](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fkaiku03)%5C **Post date:** [June 26, 2023, 3:50am UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Ftrainingargument-does-not-work-on-colab%2F43372%2F8 "2023-06-26T03:50:00Z")
Has this concern been resolved? ---
**Author:** ![Varshitha](https://avatars.discourse-cdn.com/v4/letter/v/f0a364/32.png) [@Varshitha](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2FVarshitha)%5C **Post date:** [July 6, 2023, 4:42am UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Ftrainingargument-does-not-work-on-colab%2F43372%2F9 "2023-07-06T04:42:52Z")
Thanks for this, it worked for me. ---
**Author:** ![NeMesIss](https://sea2.discourse-cdn.com/hellohellohello/user_avatar/discuss.huggingface.co/nemesiss/32/18942_2.png) [@NeMesIss](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2FNeMesIss)%5C **Post date:** [August 28, 2023, 1:13pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Ftrainingargument-does-not-work-on-colab%2F43372%2F10 "2023-08-28T13:13:15Z")
Yess worked for me too! Thanks, ---
**Author:** ![Rahi11Anurag](https://avatars.discourse-cdn.com/v4/letter/r/9d8465/32.png) [@Rahi11Anurag](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2FRahi11Anurag)%5C **Post date:** [September 22, 2023, 7:04am UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Ftrainingargument-does-not-work-on-colab%2F43372%2F11 "2023-09-22T07:04:48Z")
> [@MatthewVedder](#): > > pip install accelerate -U Thanks it worked but can you please explain this issue. ---
**Author:** ![locpham2000th](https://sea2.discourse-cdn.com/hellohellohello/user_avatar/discuss.huggingface.co/locpham2000th/32/20019_2.png) [@locpham2000th](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Flocpham2000th)%5C **Post date:** [October 3, 2023, 7:45am UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Ftrainingargument-does-not-work-on-colab%2F43372%2F12 "2023-10-03T07:45:56Z")
Thank you. Because of the above workaround I fixed the error ---
**Author:** ![jiangzizi](https://avatars.discourse-cdn.com/v4/letter/j/5f8ce5/32.png) [@jiangzizi](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fjiangzizi)%5C **Post date:** [November 11, 2023, 10:08am UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Ftrainingargument-does-not-work-on-colab%2F43372%2F13 "2023-11-11T10:08:21Z")
Also works for me! Thank you! ---
**Author:** ![Senura](https://sea2.discourse-cdn.com/hellohellohello/user_avatar/discuss.huggingface.co/senura/32/22687_2.png) [@Senura](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2FSenura)%5C **Post date:** [January 4, 2024, 4:01pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Ftrainingargument-does-not-work-on-colab%2F43372%2F14 "2024-01-04T16:01:58Z")
This works in colab. Thanks a lot and greatly appreciate you! ---
**Author:** ![bryanscheetz](https://avatars.discourse-cdn.com/v4/letter/b/c89c15/32.png) [@bryanscheetz](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fbryanscheetz)%5C **Post date:** [January 19, 2024, 4:08am UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Ftrainingargument-does-not-work-on-colab%2F43372%2F15 "2024-01-19T04:08:57Z")
This resolved the issue. Thanks! ---
**Author:** ![samman44](https://avatars.discourse-cdn.com/v4/letter/s/c4cdca/32.png) [@samman44](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fsamman44)%5C **Post date:** [February 20, 2024, 9:45pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Ftrainingargument-does-not-work-on-colab%2F43372%2F16 "2024-02-20T21:45:57Z")
Worked for me, thank you! ---
**Author:** ![mohamedhassan218](https://avatars.discourse-cdn.com/v4/letter/m/dec6dc/32.png) [@mohamedhassan218](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fmohamedhassan218)%5C **Post date:** [May 4, 2024, 9:20am UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Ftrainingargument-does-not-work-on-colab%2F43372%2F17 "2024-05-04T09:20:08Z")
Thank U for sharing ✨ ---
**Author:** ![Ansgar02](https://avatars.discourse-cdn.com/v4/letter/a/7ab992/32.png) [@Ansgar02](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2FAnsgar02)%5C **Post date:** [June 5, 2024, 4:04pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Ftrainingargument-does-not-work-on-colab%2F43372%2F18 "2024-06-05T16:04:32Z")
Thank you so much, it worked like a charm ---
**Author:** ![ElevenHu](https://sea2.discourse-cdn.com/hellohellohello/user_avatar/discuss.huggingface.co/elevenhu/32/28079_2.png) [@ElevenHu](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2FElevenHu)%5C **Post date:** [June 11, 2024, 7:40am UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Ftrainingargument-does-not-work-on-colab%2F43372%2F19 "2024-06-11T07:40:56Z")
is very nice idea ---
**Author:** ![babesultana](https://avatars.discourse-cdn.com/v4/letter/b/839c29/32.png) [@babesultana](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fbabesultana)%5C **Post date:** [May 6, 2025, 3:18am UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Ftrainingargument-does-not-work-on-colab%2F43372%2F20 "2025-05-06T03:18:39Z")
I am using this, but for this code training\_args = TrainingArguments( ``` output_dir="./results_banglabert_tier2_tier3", per_device_train_batch_size=8, per_device_eval_batch_size=8, num_train_epochs=3, logging_dir="./logs_banglabert_tier2_tier3", logging_steps=10, evaluation_strategy="epoch", save_strategy="epoch", load_best_model_at_end=True, metric_for_best_model="accuracy" ``` ) I am facing this error TypeError Traceback (most recent call last) in \() 151 # Training Arguments 152 # ======================= → 153 training\_args = TrainingArguments( 154 155 output\_dir=“./results\_banglabert\_tier2\_tier3”, TypeError: TrainingArguments. **init** () got an unexpected keyword argument ‘evaluation\_strategy’ ---
**Author:** ![John6666](https://sea2.discourse-cdn.com/hellohellohello/user_avatar/discuss.huggingface.co/john6666/32/27664_2.png) [@John6666](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2FJohn6666)%5C **Post date:** [May 6, 2025, 3:46am UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Ftrainingargument-does-not-work-on-colab%2F43372%2F21 "2025-05-06T03:46:04Z")
`evaluation_strategy` has been removed. ```py #evaluation_strategy="epoch", eval_strategy="epoch", ``` [Next page](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Ftrainingargument-does-not-work-on-colab%2F43372.md%3Fpage%3D2)