search

lhc1224 / Cross-View-AG

Official PyTorch Implementation of Learning Affordance Grounding from Exocentric Images, CVPR 2022
MIT License
47 stars 2 forks source link

TypeError: MODEL() got an unexpected keyword argument 'align_channels' #5

Closed ccmCCMfk closed 1 year ago

ccmCCMfk commented 1 year ago

when I run test.py,a typeError came out,and I found that in model.py fucion MODEL is: def MODEL(args, num_classes=36, align_w=0.2, distil_w=0.1, cls_w=1, pretrained=True,n=3,D=512): but in test.py, model = MODEL(args, num_classes=len(aff_list), align_channels=256, align_w=0.5, distil_w=0.5, cls_w=1, pretrained=False).cuda() is the patameter align_channels irrelevant? shall we just delete it? It had a result on pycharm IDE ,but not work on the server unless I delete that parameter. By the way,can I ask for the code about plotting,I'm also a student from department of automation,ustc.

ccmCCMfk commented 1 year ago

I saw some information in the extension,but do not know how to utilize it.

lhc1224 commented 1 year ago

You can email me and add me on WeChat