Traceback (most recent call last):
File "train.py", line 695, in
main(opt)
File "train.py", line 591, in main
train(opt.hyp, opt, device, callbacks)
File "train.py", line 126, in train
model = Model(cfg or ckpt['model'].yaml, ch=3, nc=nc, anchors=hyp.get('anchors')).to(device) # create
File "/remote-home/2210975/project/yoloair/models/yolo.py", line 106, in init
self.model, self.save = parse_model(deepcopy(self.yaml), ch=[ch]) # model, savelist
File "/remote-home/2210975/project/yoloair/models/yolo.py", line 341, in parse_model
m = eval(m) if isinstance(m, str) else m # eval strings
File "", line 1, in
NameError: name 'GAM_Attention' is not defined
Traceback (most recent call last): File "train.py", line 695, in
main(opt)
File "train.py", line 591, in main
train(opt.hyp, opt, device, callbacks)
File "train.py", line 126, in train
model = Model(cfg or ckpt['model'].yaml, ch=3, nc=nc, anchors=hyp.get('anchors')).to(device) # create
File "/remote-home/2210975/project/yoloair/models/yolo.py", line 106, in init
self.model, self.save = parse_model(deepcopy(self.yaml), ch=[ch]) # model, savelist
File "/remote-home/2210975/project/yoloair/models/yolo.py", line 341, in parse_model
m = eval(m) if isinstance(m, str) else m # eval strings
File "", line 1, in
NameError: name 'GAM_Attention' is not defined