ffcv-imagenet icon indicating copy to clipboard operation
ffcv-imagenet copied to clipboard

Published 20 hours ago • verified publisher icon libffcv

resolve deprecated compute_on_step

Open h2th3k opened this issue 9 months ago • 0 comments

Hi, Start from here compute_on_step is deprecated. New version of torchmetrics will rase ValueError: Unexpected keyword arguments: compute_on_step.

Quick fix is to remove them to be compatible with the new version(1.4.0): e.g. https://github.com/libffcv/ffcv-imagenet/blob/4dd291a122096a2032ec9fd19633e1b9aff3577d/train_imagenet.py#L419

h2th3k avatar May 10 '24 18:05 h2th3k