Training errors on LIP

[abhigoku10]

@PeikeLi @usernameandme thanks for the sharing the code base , i am getting the follwoing error when i am trying to replicated the results for the LIP dataset
"Traceback (most recent call last):
File "train.py", line 233, in
main()
File "train.py", line 190, in main
soft_edge.append(soft_pred[1][-1])
IndexError: list index out of range"

from schp_1_checkpoint.pth.tar during the cycle training i am getting this error

[greatbaozi001]

Hello, I have met the same error, have you figured it out?

[KudoKhang]

You can try my code!

[greatbaozi001]

You can try my code!

thanks for your reply! I'll try it later. : )

[duACGN]

You can try my code!

你好，我能问一个简单的问题吗，请问您知道作者这里原代码的真正用意是什么吗，我看你改的代码只适用于batch size = 2 的情况
Hello, can I ask you a simple question? Do you know what the author's original code really means? I think the code you changed is only applicable to batch size=2