Получите пакет ускорения для регистрации результатов тестов с помощью Huggingface Trainer. ⇐ Python

[Anonymous]

Я настраиваю модель T5 для конкретного набора данных, и мой код выглядит следующим образом:

accelerator = Accelerator(log_with='wandb') tokenizer = T5Tokenizer.from_pretrained('t5-base') модель = T5ForConditionalGeneration.from_pretrained('t5-base') Accelerator.init_trackers( имя_проекта = 'мойПроект', конфигурация={ # Мои конфиги } ) # Затем я делаю некоторые приготовления к тонкой настройке Trainer_arguments = Transformers.Seq2SeqTrainingArguments( # Здесь я передаю много аргументов ) тренер = трансформеры.Seq2SeqTrainer( # Здесь я передаю аргументы вместе с другими необходимыми аргументами ) # ЗАТЕМ НАКОНЕЦ Я ТРЕНИРУЮ, ОЦЕНИВАЮ И ТЕСТИРУЮ ТАК: тренер.поезд() Trainer.evaluate(#параметры оценки#) тренер.predict(#тестовые аргументы#) Теперь моя основная проблема: когда я проверяю сайт wandb для своего проекта, я вижу только журналирование фазы trainer.train(), но не Trainer.evaluate() или trainer.predict().

Я порылся в Интернете, пытаясь найти решение, но не нашел его.
Как мне заставить wandb/accelerate регистрировать все мои фазы?
Спасибо!