Closed segalinc closed 1 year ago
@segalinc could you pls chare more details like what package versions you are using?
torchmetrics @ git+https://github.com/Lightning-AI/torchmetrics.git@0d7d6c7ccfedbb4941dfc55e65bbfca88506b2dc
I also tried with version 1.1.12 and 1.2.0
I am using python not conda
I assume it's because you do not have torch-fidelity installed and it therefore is a dummy class that is trying to be initialized which does only take one argument. @segalinc could you please confirm if this is true? @Borda we should probably guard against this in some way
Yes installing torch-fidelity solved the problem Maybe adding this note in the readme.md or as requirements can help thank you
Hello,
trying to init NoTrainInceptionV3 with latest version
but I get