computervision-recipes icon indicating copy to clipboard operation
computervision-recipes copied to clipboard

Exception: It's not possible to apply those transforms to your dataset: expected scalar type Float but found Double

Open curryJ opened this issue 3 years ago • 1 comments

Description

Thanks for this great project. I run the 02_state_of_the_art.ipynb,but I get a Exception which is the title; My computer is cuda10.1 and GPU version is rtx2080ti.I update the version of pytorch and torchvision to 1.7.0 and 0.8.0 respectively

curryJ avatar Jun 20 '21 08:06 curryJ

Looks like a version conflict. You could try this conda environment, that should work: https://github.com/microsoft/computervision-recipes/blob/master/environment.yml

PatrickBue avatar Jun 20 '21 20:06 PatrickBue