src/benchmark.py
def to(self, device):
        '''
        Move model to device specified by parameter
        '''
        self._device = device
        self.tree_lstm.to(device)
        self.embeddings.to(device)
        self.out.to(device)
        return self

        Move model to device specified by parameter
        
Move model to device specified by parameter
Move model to device specified by parameter
def to(self, device):
        self._device = device
        self.tree_lstm.to(device)
        self.embeddings.to(device)
        self.out.to(device)
        return self
def to ( self , device ) : self . _device = device self . tree_lstm . to ( device ) self . embeddings . to ( device ) self . out . to ( device ) return self
Move model to device specified by parameter
Move model to device specified by parameter
'''
        Move model to device specified by parameter
        '''
self
device