Skip to content

S
SENSE-PPI
Commit b73cbabe by Konstantin Volzhenin
Options

0.2.0 minor bugfix in tqdm

parent b3efe4de
Showing with 2 additions and 2 deletions
senseppi/commands/train.py
import pytorch_lightning as pl import pytorch_lightning as pl
from pytorch_lightning.callbacks import ModelCheckpoint from pytorch_lightning.callbacks import ModelCheckpoint, TQDMProgressBar
import pathlib import pathlib
import argparse import argparse
from ..utils import * from ..utils import *
...@@ -29,7 +29,7 @@ def main(params): ...@@ -29,7 +29,7 @@ def main(params):
logger = None logger = None
callbacks = [ callbacks = [
# TQDMProgressBar(refresh_rate=250), TQDMProgressBar(refresh_rate=50),
ModelCheckpoint(filename='chkpt_loss_based_{epoch}-{val_loss:.3f}-{val_BinaryF1Score:.3f}', verbose=True, ModelCheckpoint(filename='chkpt_loss_based_{epoch}-{val_loss:.3f}-{val_BinaryF1Score:.3f}', verbose=True,
monitor='val_loss', mode='min', save_top_k=1) monitor='val_loss', mode='min', save_top_k=1)
] ]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment