使用BISRNRT作为Net训练模型时出现问题 about bi-srnet HOT 5 OPEN

Thank you, the loss function is nan because my batchsize is not divisible, I set Droplast to True and can train normally! There is actually a minor problem. I noticed that both the 'Models' folder and the main folder have a file named bisrnet.py, which defines the head layer, and the former does not. What is the purpose of these two? Thank you again! 在2022年5月23日 ***@***.***> 写道： Hi. Is the batch size too small? The BiSRNet is a bit hard to train but the loss shouldn't be crazy. I got accuracy improvements while freezing the other model parts and training only the SR modules. You can also try that. — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: ***@***.***>