Run train.py ERROR

[kaikaigood]

Hi，foamliu：

1. When “loss=overall_loss”, "pretrained_path = None" and "num_gpu=1" are set, an error occurs when running “train.py”. The error situation is as follows：

• code: final.compile(optimizer='nadam', loss=overall_loss, target_tensors=[decoder_target])
• error: Traceback (most recent call last):
  File "C:\Users\Administrator\anaconda3\envs\tensorflow1.15\lib\site-packages\tensorflow_core\python\client\session.py", line 1365, in _do_call
  return fn(*args)
  File "C:\Users\Administrator\anaconda3\envs\tensorflow1.15\lib\site-packages\tensorflow_core\python\client\session.py", line 1350, in _run_fn
  target_list, run_metadata)
  File "C:\Users\Administrator\anaconda3\envs\tensorflow1.15\lib\site-packages\tensorflow_core\python\client\session.py", line 1443, in _call_tf_sessionrun
  run_metadata)
  tensorflow.python.framework.errors_impl.InvalidArgumentError: 2 root error(s) found.
  (0) Invalid argument: Incompatible shapes: [8,320,320,0] vs. [0,320,320,3]
  [[{{node loss/refinement_pred_loss/sub_2}}]]
  [[loss/mul/_1021]]
  (1) Invalid argument: Incompatible shapes: [8,320,320,0] vs. [0,320,320,3]
  [[{{node loss/refinement_pred_loss/sub_2}}]]
  0 successful operations.
  0 derived errors ignored.

2. When “loss=alpha_prediction_loss”is set, Training can run normally.

3. overall_loss=0.5* alpha_prediction_loss(y_true, y_pred) + 0.5* compositional_loss(y_true, y_pred), so I doubt whether it is caused by the problem of compositional_loss.

I wonder if you have encountered this problem during training?

Thank you!