multi-class trained error

Question

Closed this issue 3 years ago · 2 comments

Thanks for your good work，if this code just for one class，when i used to train mutil-classs model，i got error

Answer 1 · 2021-09-22T09:13:14.000Z

Thanks for attention. For multi-class cases, we use the pipeline of TransUNet: https://github.com/Beckschen/TransUNet, hope this helps.

Answer 2 · 2021-09-23T08:15:25.000Z

thank you for your response, I really appreciate it，the mutil-class is trained successfully.