File tree Expand file tree Collapse file tree
examples/references/classification/imagenet/code/utils Expand file tree Collapse file tree Original file line number Diff line number Diff 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
7676def _mlflow_get_output_path ():
Original file line number Diff line number Diff 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__,
You can’t perform that action at this time.
0 commit comments