Compatibility issues with latest pytorch

Oulu-IMEDS / solt

Streaming over lightweight data transformations

https://oulu-imeds.github.io/solt/

MIT License

263 stars 19 forks source link

Compatibility issues with latest pytorch #74

Closed soupault closed 2 years ago

soupault commented 4 years ago

Works fine with 1.1, but not 1.5. Traceback:

    as_dict=as_dict, scale_keypoints=scale_keypoints, normalize=normalize, mean=mean, std=std,
  File ".../site-packages/solt/core/_data.py", line 273, in to_torch
    img.sub_(mean)
RuntimeError: output with backend CPU and dtype Byte doesn't match the desired backend CPU and dtype Float
Process finished with exit code 1

lext commented 3 years ago

This does not look right. I am using 1.8 version.