wwMark / meshgraphnets

Rewrite deepmind/meshgraphnets into pytorch
75 stars 23 forks source link

TypeError: forward() got multiple values for argument 'is_training' #9

Open nothingtoexplain opened 2 years ago

nothingtoexplain commented 2 years ago

I encountered a problem after running the run_model.py

The erro is File "/home/zhn/zha/code/meshgraphnets-main/run_model.py", line 1084, in <module> app.run(main) File "/home/zhn/anaconda3/envs/meshgraphnets/lib/python3.9/site-packages/absl/app.py", line 312, in run _run_main(main, args) File "/home/zhn/anaconda3/envs/meshgraphnets/lib/python3.9/site-packages/absl/app.py", line 258, in _run_main sys.exit(main(argv)) File "/home/zhn/zha/code/meshgraphnets-main/run_model.py", line 766, in main train_loss_record = learner(model, params, run_step_config) File "/home/zhn/zha/code/meshgraphnets-main/run_model.py", line 364, in learner network_output = model(inputs=data_frame, is_training=True) File "/home/zhn/anaconda3/envs/meshgraphnets/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1051, in _call_impl return forward_call(*input, **kwargs) File "/home/zhn/zha/code/meshgraphnets-main/cfd_model.py", line 127, in forward return self.learned_model(graph, self._edge_normalizer, is_training=True) File "/home/zhn/anaconda3/envs/meshgraphnets/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1051, in _call_impl return forward_call(*input, **kwargs) TypeError: forward() got multiple values for argument 'is_training' I haven't changed the code yet when I encountered the above problem. I hope someone can help me. Thanks a lot.