前提
環境 Windows10
Jupyterlab
pytorch_lightningで画像処理を実装をしています。
verboseをfalseにしているにもかかわらず、複数に重複して表示されます。
(1 epochに対し、複数のプログレスバーが表示されます、、、)
実現したいこと
1 epochに対し、プログレスバーを1行の表示で行いたいのですが、どうやったらできるのでしょうか?
- ▲▲機能を動作するようにする
発生している問題・エラーメッセージ
##出力 data_module <__main__.MyDataModule object at 0x7f2bb2d73790> Epoch 0: 86%|████████▌ | 25/29 [00:33<00:05, 1.34s/it, loss=0.603, v_num=68] Validating: 0it [00:00, ?it/s] Validating: 0%| | 0/4 [00:00<?, ?it/s] Epoch 0: 93%|█████████▎| 27/29 [00:34<00:02, 1.29s/it, loss=0.603, v_num=68] Validating: 50%|█████ | 2/4 [00:01<00:01, 1.59it/s] Epoch 0: 100%|██████████| 29/29 [00:35<00:00, 1.22s/it, loss=0.603, v_num=68] Epoch 0: 100%|██████████| 29/29 [00:35<00:00, 1.23s/it, loss=0.603, v_num=68] Epoch 0: 100%|██████████| 29/29 [00:35<00:00, 1.23s/it, loss=0.603, v_num=68] Trainer was signaled to stop but required minimum epochs (2) or minimum steps (None) has not been met. Training will continue... Epoch 1: 90%|████████▉ | 26/29 [00:31<00:03, 1.20s/it, loss=0.503, v_num=68] Validating: 0it [00:00, ?it/s] Validating: 0%| | 0/4 [00:00<?, ?it/s] Validating: 25%|██▌ | 1/4 [00:01<00:03, 1.12s/it] Epoch 1: 97%|█████████▋| 28/29 [00:32<00:01, 1.17s/it, loss=0.503, v_num=68] Validating: 75%|███████▌ | 3/4 [00:01<00:00, 2.06it/s] Epoch 1: 100%|██████████| 29/29 [00:33<00:00, 1.15s/it, loss=0.503, v_num=68] Epoch 1: 100%|██████████| 29/29 [00:33<00:00, 1.15s/it, loss=0.503, v_num=68]
該当のソースコード
python
1early_stop_callback = EarlyStopping( 2 monitor='val_loss', 3 min_delta=0.05, 4 patience=3, 5 mode='min' 6 7) 8 9checkpoint_callback = ModelCheckpoint( 10 filename='{epoch}', 11 monitor='val_loss', 12 mode='min', 13 # verbose=True, 14 verbose=False 15) 16 17trainer = pl.Trainer( 18 max_epochs=20, 19 gpus=1, 20 callbacks=[checkpoint_callback, early_stop_callback], 21 log_every_n_steps=10) 22 23trainer.fit(model, data_module)
試したこと
verbose=Falseをearly_stop_callback、checkpoint_callbackの二つにも適用したのですが、複数行の表示になりました。
補足情報(FW/ツールのバージョンなど)
torch 1.10.1+cu111
python-dateutil 2.8.2
python-json-logger 2.0.1
pytorch-lightning 1.5.9
回答1件
あなたの回答
tips
プレビュー