Confusion about the code in net/resnet50_amr.py

[Jessie459]

Hello! Thanks for your great work!

I am confused about some lines in net/resnet50_amr.py. Did you mean that the spotlight branch is not supervised by classification? Is this equal to with torch.no_grad()? https://github.com/JayQine/AMR/blob/de391682626990deebe8eb217b7eed6ec4d7fd3d/net/resnet50_amr.py#L47-L58

Looking forward to your reply.

[JayQine]

There are two training stages. The first trains the spotlight branch, and the second trains compensation branch in xxx_amr.py. In the second stage, some parameters of spotlight branch are fixed.