Skip to content

ValueError: Unexpected keyword arguments: compute_on_step #14

@LYJ0327

Description

@LYJ0327

Hi, Hello, I'm very sorry to bother you, but I would like to ask why my program reports the following error whether I run test or train:
Traceback (most recent call last):
File "/home/cvt2/bin/dlhpcstarter", line 8, in
sys.exit(main())
^^^^^^
File "/home/cvt2/lib/python3.11/site-packages/dlhpcstarter/main.py", line 54, in main
submit(args=args, stages_fnc=stages_fnc)
File "/home/cvt2/lib/python3.11/site-packages/dlhpcstarter/main.py", line 69, in submit
stages_fnc(args)
File "/home/cvt2distilgpt2/stages.py", line 68, in stages
model = TaskModel(**vars(args))
^^^^^^^^^^^^^^^^^^^^^^^
File "/home/cvt2distilgpt2/cvt2distilgpt2_iu_x_ray_chen.py", line 66, in init
self.val_coco_metrics = COCOCaptionMetrics(metrics=["bleu", "cider", "rouge"])
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/cvt2distilgpt2/tools/metrics/coco.py", line 33, in init
super().init(dist_sync_on_step=dist_sync_on_step, compute_on_step=False)
File "/home/cvt2/lib/python3.11/site-packages/torchmetrics/metric.py", line 146, in init
raise ValueError(f"Unexpected keyword arguments: {', '.join(kwargs_)}")
ValueError: Unexpected keyword arguments: compute_on_step
I would be very grateful if you could reply me with a solution to my problem!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions