ShuaiBai623's picture lim4349's picture
Fix: move inputs to model device in inference example to avoid "same device" error (#3)
b7e6648 verified