Python use_cuda Exemples

Exemple #1

Fichier : genre_lstm.py Projet : jarmand-caw/Streaming_AI

 def __init__(self, config):
     self.model = GenreNet(config)
     if config['use_cuda'] is True:
         use_cuda(True)
         self.model.cuda()
     super(GenreEngine, self).__init__(config)
     print(self.model)

Exemple #2

Fichier : dlrm.py Projet : jarmand-caw/Streaming_AI

    def __init__(self, config):
        self.model = DLRM(config)
        if config['use_cuda'] is True:
            use_cuda(True)
            self.model.cuda()
        super(DLRMEngine, self).__init__(config)
        print(self.model)

        if config['pretrain']:
            self.model.load_pretrain_weights()

Exemple #3

Fichier : matrix_factorization.py Projet : jarmand-caw/Streaming_AI

 def __init__(self, config):
     self.model = GMF(config)
     if config['use_cuda'] is True:
         use_cuda(True)
         self.model.cuda()
     super(GMFEngine, self).__init__(config)