The error in pretrain.py

[WangHonghui123]

Thank you for your nice work. When I run your code, I find that there is an error, shown as:

preTrainer = Trainer(logger, callbacks = callbacks, checkpoint_callback=False, num_sanity_val_steps=0, progress_bar_refresh_rate=10, cfg.trainer)

The error information is shown as follows:

Traceback (most recent call last): File "/Users/wanghonghui/PycharmProjects/Trajectory_Prediction/venv/lib/python3.8/site-packages/hydra/_internal/utils.py", line 394, in _run_hydra _run_app( File "/Users/wanghonghui/PycharmProjects/Trajectory_Prediction/venv/lib/python3.8/site-packages/hydra/_internal/utils.py", line 457, in _run_app run_and_report( File "/Users/wanghonghui/PycharmProjects/Trajectory_Prediction/venv/lib/python3.8/site-packages/hydra/_internal/utils.py", line 223, in run_and_report raise ex File "/Users/wanghonghui/PycharmProjects/Trajectory_Prediction/venv/lib/python3.8/site-packages/hydra/_internal/utils.py", line 220, in run_and_report return func() File "/Users/wanghonghui/PycharmProjects/Trajectory_Prediction/venv/lib/python3.8/site-packages/hydra/_internal/utils.py", line 458, in lambda: hydra.run( File "/Users/wanghonghui/PycharmProjects/Trajectory_Prediction/venv/lib/python3.8/site-packages/hydra/_internal/hydra.py", line 132, in run _ = ret.return_value File "/Users/wanghonghui/PycharmProjects/Trajectory_Prediction/venv/lib/python3.8/site-packages/hydra/core/utils.py", line 260, in return_value raise self._return_value File "/Users/wanghonghui/PycharmProjects/Trajectory_Prediction/venv/lib/python3.8/site-packages/hydra/core/utils.py", line 186, in run_job ret.return_value = task_function(task_cfg) File "/Users/wanghonghui/PycharmProjects/Trajectory_Prediction/GoalGAN-master/run.py", line 50, in run pretrain_func(generator=model.generator, File "/Users/wanghonghui/PycharmProjects/Trajectory_Prediction/GoalGAN-master/model/pretrain_pl.py", line 26, in pretrain_func preTrainer = Trainer(logger, File "/Users/wanghonghui/PycharmProjects/Trajectory_Prediction/venv/lib/python3.8/site-packages/pytorch_lightning/utilities/argparse.py", line 69, in insert_env_defaults return fn(self, **kwargs) TypeError: init() got an unexpected keyword argument 'checkpoint_callback' python-BaseException

I would like to know if you can fix it. Thank you for your help.

[dendorferpatrick]

Hi, I have not yet encountered this problem. Can you check, if you are using the supported torch version torch==1.6.0.