predict_emotion.py 文件源码

def __set_cpu_or_gpu(self):
        # Prepare multi-layer perceptron model, defined in net.py
        if self.gpu >= 0:
            cuda.get_device(self.gpu).use()
            self.model.to_gpu()
        self.xp = np if self.gpu < 0 else cuda.cupy