评价指标报错

[zhoulian-sudo]

您好，周梓康博士，感谢您杰出的代码贡献，想问下您评价指标处一直报这种错误是什么原因？？
报错处：class ADE(Metric):

def __init__(self,
             compute_on_step: bool = True,
             dist_sync_on_step: bool = False,
             process_group: Optional[Any] = None,
             dist_sync_fn: Callable = None) -> None:
    super(ADE, self).__init__(compute_on_step=compute_on_step, dist_sync_on_step=dist_sync_on_step,
                              process_group=process_group, dist_sync_fn=dist_sync_fn)
    self.add_state('sum', default=torch.tensor(0.0), dist_reduce_fx='sum')
    self.add_state('count', default=torch.tensor(0), dist_reduce_fx='sum')

错误：
Global seed set to 2022
/home/amax/anaconda3/envs/haha/lib/python3.9/site-packages/pytorch_lightning/trainer/connectors/accelerator_connector.py:466: LightningDeprecationWarning: Setting Trainer(gpus=1) is deprecated in v1.7 and will be removed in v2.0. Please use Trainer(accelerator='gpu', devices=1) instead.
rank_zero_deprecation(
GPU available: True (cuda), used: True
TPU available: False, using: 0 TPU cores
IPU available: False, using: 0 IPUs
HPU available: False, using: 0 HPUs
Traceback (most recent call last):
File "/data/project/HiVT-main/train.py", line 44, in
model = HiVT(**vars(args))
File "/data/project/HiVT-main/models/hivt.py", line 86, in init
self.minADE = ADE()
File "/data/project/HiVT-main/metrics/ade.py", line 27, in init
super(ADE, self).init(compute_on_step=compute_on_step, dist_sync_on_step=dist_sync_on_step,
File "/home/amax/anaconda3/envs/haha/lib/python3.9/site-packages/torchmetrics/metric.py", line 145, in init
raise ValueError(f"Unexpected keyword arguments: {', '.join(kwargs_)}")
ValueError: Unexpected keyword arguments: compute_on_step

[LL-C8]

@zhoulian-sudo 同样遇到了这个问题，请问解决了吗？

[zhoulian-sudo]

不好意思，我这边也没有解决  冰糖?葫芦 ***@***.***  

…

------------------ 原始邮件 ------------------ 发件人: ***@***.***>; 发送时间: 2023年8月6日(星期天) 下午4:22 收件人: ***@***.***>; 抄送: ***@***.***>; ***@***.***>; 主题: Re: [ZikangZhou/HiVT] 评价指标报错 (Issue #41) @zhoulian-sudo 同样遇到了这个问题，请问解决了吗？ — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you were mentioned.Message ID: ***@***.***>

[Lewis-Lu]

Hi there,

Could you please check the version of torchmetrics using pip list | grep torchmetrics.

As far as I know, Metrics has deprecate the compute_on_step parameter from certain version.

Thanks
Lewis

[xuyangwenhui]

I solved the same problem by using commod " pip install torchmetrics==0.8.2 ", maybe you could reference the issue " Lightning-AI/torchmetrics#789 "

[zhaozhen2333]

I solved the same problem by using commod " pip install torchmetrics==0.8.2 ", maybe you could reference the issue " Lightning-AI/torchmetrics#789 "

Thank you very much

[alien-HL]

把3个评价指标的文件中，super初始化里的compute_on_step关键字去掉就可以