line 333: fixed next_state.device = 'cuda'

libffcv / ffcv

FFCV: Fast Forward Computer Vision (and other ML workloads!)

https://ffcv.io

Apache License 2.0

2.84k stars 178 forks source link

Closed gthampak closed 1 year ago

gthampak commented 1 year ago

originally, next_state.device.type = 'cuda'; removed .type

andrewilyas commented 1 year ago

Hi @gthampak ! As of the latest version of FFCV, device is meant to be a torch.device not a string, sorry about the confusion!