colab cpu inference

[AK391]

great work, is it possible to do inference on cpu in colab?

[mchong6]

Added device which lets you use cpu. But runs out of memory on colab if doing too many images so have to be careful for that.

[AK391]

@mchong6 thanks the projection part still requires gpu?

[mchong6]

Sorry, just fixed that!