Hi I am pretty new to CNNs, I downloaded the following code by Aman Kumar Mallik and only changed the variable device from "cuda" to "cpu" (I cannot use &quo