LoraLinH / Gramformer

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Are there problems with parameter Settings for regression_trianer file?

jizhuangz opened this issue · comments

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!

commented

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