AttributeError: module 'pytorch_lightning' has no attribute 'callbacks'の原因が分かりません
pytorch初心者です。
pytorch_ligthtningを使い、BERTをファインチューニングしたいのですがうまくいきません。Googleコラボ上ではこのようなエラーは出ませんでしたが、jupyter notebook上で同じコードを実行するとエラーが発生します。
発生している問題・エラーメッセージ
AttributeError Traceback (most recent call last) <ipython-input-18-8c934190102a> in <module> 1 # 学習時にモデルの重みを保存する条件を指定 ----> 2 checkpoint = pl.callbacks.ModelCheckpoint( 3 monitor='val_loss', 4 mode='min', 5 save_top_k=1, AttributeError: module 'pytorch_lightning' has no attribute 'callbacks'
該当のソースコード
import random import glob from tqdm import tqdm import torch from torch.utils.data import DataLoader from transformers import BertJapaneseTokenizer, BertForSequenceClassification import pytorch_lightning as pl MODEL_NAME = 'cl-tohoku/bert-base-japanese-whole-word-masking' tokenizer = BertJapaneseTokenizer.from_pretrained(MODEL_NAME) bert_sc = BertForSequenceClassification.from_pretrained( MODEL_NAME, num_labels=5 ) bert_sc = bert_sc.cuda() . . . # 学習時にモデルの重みを保存する条件を指定 checkpoint = pl.callbacks.ModelCheckpoint(#エラー monitor='val_loss', mode='min', save_top_k=1, save_weights_only=True, dirpath="/tabuchi", ) # 学習の方法を指定 trainer = pl.Trainer( gpus=1, max_epochs=10, callbacks = [checkpoint] )
補足情報(FW/ツールのバージョンなど)
pytorch-lightning==1.2.10
pytorch-lightningのモジュールを調べたところ、コラボ、jupyter notebook上それぞれ以下のようになっていました。
コラボ↓
['Callback',
'LightningDataModule',
'LightningModule',
'Trainer',
'_PACKAGE_ROOT',
'_PROJECT_ROOT',
'all',
'author',
'author_email',
'builtins',
'cached',
'copyright',
'doc',
'docs',
'file',
'homepage',
'license',
'loader',
'name',
'package',
'path',
'spec',
'version',
'_logger',
'_root_logger',
'accelerators',
'callbacks',
'core',
'distributed',
'info',
'loggers',
'logging',
'metrics',
'os',
'overrides',
'plugins',
'profiler',
'seed_everything',
'trainer',
'tuner',
'utilities']
jupyter notebook↓
['Callback',
'LightningDataModule',
'LightningModule',
'Trainer',
'_PACKAGE_ROOT',
'_PROJECT_ROOT',
'all',
'author',
'author_email',
'builtins',
'cached',
'copyright',
'doc',
'docs',
'file',
'homepage',
'license',
'loader',
'name',
'package',
'path',
'spec',
'version',
'_logger',
'_root_logger',
'logging',
'metrics',
'os',
'seed_everything']
同じバージョンをインストールしいるはずなのに、モジュールが無いです。
回答2件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。