Are there problems with parameter Settings for regression_trianer file?

LoraLinH / Gramformer

Official Implement of AAAI 2024 paper 'Gramformer: Learning Crowd Counting via Graph-Modulated Transformer'

MIT License

12 stars 1 forks source link

Are there problems with parameter Settings for regression_trianer file? #1

Open jizhuangz opened 1 month ago

jizhuangz commented 1 month ago

regression_trianer file line 66, self.model = vgg19_trans(args.topk, args.usenum, args.promptnum), promptunm is not defined in train.py file. AttributeError: 'Namespace' object has no attribute 'promptnum' regression_trianer file line 89,self.save_list = Save_Handle(max_num=args.max_model_num), max_model_num is not defined in train.py file too.How to set these two values? Thanks in advance for your help!

LoraLinH commented 1 month ago

Thanks for the reminder. I've made the corresponding revisions.