Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update progess bar

This commit is contained in:
shahules786 2022-09-29 10:36:47 +05:30
parent 31a3335ff0
commit 206355270e
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
cli/train.py
View File

@ -3,11 +3,14 @@ from hydra.utils import instantiate
from omegaconf import DictConfig from omegaconf import DictConfig
from pytorch_lightning.callbacks import ModelCheckpoint, EarlyStopping from pytorch_lightning.callbacks import ModelCheckpoint, EarlyStopping
from pytorch_lightning.loggers import MLFlowLogger from pytorch_lightning.loggers import MLFlowLogger
from pytorch_lightning.callbacks import TQDMProgressBar
@hydra.main(config_path="train_config",config_name="config") @hydra.main(config_path="train_config",config_name="config")
def main(config: DictConfig): def main(config: DictConfig):
callbacks = [] callbacks = []
callbacks.append(TQDMProgressBar(refresh_rate=10))
logger = MLFlowLogger(experiment_name=config.mlflow.experiment_name, logger = MLFlowLogger(experiment_name=config.mlflow.experiment_name,
run_name=config.mlflow.run_name) run_name=config.mlflow.run_name)