Skip to content

Commit 444353b

Browse files
Merge branch 'master' into Refactor-distributed-tests
2 parents 0483113 + b04cbfc commit 444353b

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

examples/references/classification/imagenet/code/utils/exp_tracking.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -69,8 +69,8 @@ def _plx_log_params(params_dict):
6969
from polyaxon_client.tracking import Experiment
7070

7171
plx_exp = Experiment()
72-
plx_exp.log_params(**{"pytorch version": torch.__version__, "ignite version": ignite.__version__})
73-
plx_exp.log_params(**params_dict)
72+
plx_exp.log_inputs(**{"pytorch version": torch.__version__, "ignite version": ignite.__version__})
73+
plx_exp.log_inputs(**params_dict)
7474

7575

7676
def _mlflow_get_output_path():

ignite/contrib/handlers/polyaxon_logger.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ class PolyaxonLogger(BaseLogger):
4040
plx_logger = PolyaxonLogger()
4141
4242
# Log experiment parameters:
43-
plx_logger.log_params(**{
43+
plx_logger.log_inputs(**{
4444
"seed": seed,
4545
"batch_size": batch_size,
4646
"model": model.__class__.__name__,

0 commit comments

Comments
 (0)