문제 상황
!python run_clm_no_trainer.py \\
--dataset_name wikitext \\
--dataset_config_name wikitext-2-raw-v1 \\
--model_name_or_path gpt2 \\
--output_dir /tmp/test-clm
gpt2 를 돌리던 와중에 에러 발생
해결방법
배치사이즈를 줄여보자
--per_device_train_batch_size 2
References
'TIL > Colab' 카테고리의 다른 글
[Colab] Colab IP 찾기 (0) | 2022.05.16 |
---|---|
[Colab] Mecab name 'Tagger' is not defined (0) | 2021.12.26 |
[Colab] TypeError: Cannot convert a symbolic Keras input/output to a numpy array. (0) | 2021.11.10 |
[Colab] 런타임 연결 끊김 방지 (2) | 2021.10.13 |
Google Drive + Google Colab + GitHub 연동하기 (11) | 2021.08.25 |