[quote=“ptrblck, post:2, topic:30244”]
device=device
[/quote
But the error is resolved after what you asked to add.
I am getting a size issue though
invalid argument 2: size ‘[16384]’ is invalid for input with 1376256 elements at /pytorch/aten/src/TH/THStorage.c:41
I think I can resolve it after playing around with the network. Any inputs will be greatly appreciated.
Thank you very much.