运行train.py时，提示ft_net 找不到usam

[CloudStrifer]

运行train.py时，提示ft_net 找不到usam，不知道怎么回事，请问有人知道吗？

0%| | 16/12185 [00:00<10:53, 18.61it/s]Traceback (most recent call last):
File "F:\PyCode\Person_reID_baseline_pytorch-master\train2.py", line 626, in
main()
File "F:\PyCode\Person_reID_baseline_pytorch-master\train2.py", line 621, in main
model = train_model(model, criterion, optimizer_ft, exp_lr_scheduler,
File "F:\PyCode\Person_reID_baseline_pytorch-master\train2.py", line 128, in train_model
outputs = model(inputs)
File "D:\Python310\lib\site-packages\torch\nn\modules\module.py", line 1751, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "D:\Python310\lib\site-packages\torch\nn\modules\module.py", line 1762, in _call_impl
return forward_call(*args, **kwargs)
File "F:\PyCode\Person_reID_baseline_pytorch-master\model.py", line 135, in forward
if self.usam:
File "D:\Python310\lib\site-packages\torch\nn\modules\module.py", line 1940, in getattr
raise AttributeError(
AttributeError: 'ft_net' object has no attribute 'usam'
0%| | 16/12185 [00:11<2:24:15, 1.41it/s]

[layumi]

Thank you @CloudStrifer
Sorry for the late reply. It is solved.