170 похожих чатов

Привет. Нужна помощь с FRED-T5 torch.cuda.OutOfMemoryError Добрался до статейки https://habr.com/ru/articles/751580/?amp&amp Статья про

датасет, но там простой код файнтюнинга.
Я поигрался, потюнил Fred Large на это датасете. Приколько.
Собрал датасет под свою задачу. 5млн записей.
Решил потренироваться и заодно взял модель 1.7B
У меня 3090ti
Запустил обучение на датаесете в 300 примеров с размером батча 8.
Пролетело.
Я решил, что памяти хватает.
Но вот беда, если я запускаю на полном (5млн) датасете. Даже с размером батча 2.
Падает с torch.cuda.OutOfMemoryError сразу же, как встречает trainer.train

Я не могу понять почему. Подскажите что не так?

Могу получить до кучи по башке, что не сделал стриминговый DataLoader. Но обычной оперативы у меня хватает сожрать датасет.

2 ответов

21 просмотр

Напишите параметры оьучениы ваши. Также длину контекста

Максим-Антонов Автор вопроса
datascience xc
Напишите параметры оьучениы ваши. Также длину конт...

./train.py --model_name_or_path=models --dataset_path=dataset_orig.json --output_dir=models --learning_rate=1e-02 --gradient_accumulation_steps=3 --num_train_epochs=2 --save_steps=2000 --save_total_limit=10 --optim=adafactor --seed=42 --report_to=tensorboard --logging_steps=30 --per_device_train_batch_size=2 Вот так запускал. Все эти параметры в итоге уходят в TrainerArgs Длина контекста это что? Длина самих сообщений в input? Не измерял максимальную … Но… Стоит дать датасет на все 5 млн сообщений, оно падает, как только входит в Trainer.train() Стоит обрезать датасет (взять начало) до пары миллионов - норм тренирует (на экспериментах я выжидал до пары сотен шагов и вырубал)

Похожие вопросы

Обсуждают сегодня

Господа, а что сейчас вообще с рынком труда на делфи происходит? Какова ситуация?
Rꙮman Yankꙮvsky
29
А вообще, что может смущать в самой Julia - бы сказал, что нет единого стандартного подхода по многим моментам, поэтому многое выглядит как "хаки" и произвол. Короче говоря, с...
Viktor G.
2
30500 за редактор? )
Владимир
47
а через ESC-код ?
Alexey Kulakov
29
Чёт не понял, я ж правильной функцией воспользовался чтобы вывести отладочную информацию? но что-то она не ловится
notme
18
У меня есть функция где происходит это: write_bit(buffer, 1); write_bit(buffer, 0); write_bit(buffer, 1); write_bit(buffer, 1); write_bit(buffer, 1); w...
~
14
Добрый день! Скажите пожалуйста, а какие программы вы бы рекомендовали написать для того, чтобы научиться управлять памятью? Можно написать динамический массив, можно связный ...
Филипп
7
Недавно Google Project Zero нашёл багу в SQLite с помощью LLM, о чём достаточно было шумно в определённых интернетах, которые сопровождались рассказами, что скоро всех "ибешни...
Alex Sherbakov
5
Ребят в СИ можно реализовать ООП?
Николай
33
https://github.com/erlang/otp/blob/OTP-27.1/lib/kernel/src/logger_h_common.erl#L174 https://github.com/erlang/otp/blob/OTP-27.1/lib/kernel/src/logger_olp.erl#L76 15 лет назад...
Maksim Lapshin
20
Карта сайта