Exemplo n.º 1
0
    def __init__(self, port, config, device):
        PTServer.__init__(self, port=port, config=config, device=device)
        self.config["irank"] = 0

        self.max_mb = self.config["n_epochs"] * len(self.data[0])
        print "max_mb = %d" % self.max_mb
        self.uidx = {}
        self.valid = {}
        self.adj_lr = {}
        self.start_time = None
        self.last = None
        self.last_uidx = 0
        self.validFreq = len(self.data[0])
        self.uepoch = 0
        self.last_uepoch = 0

        if self.config["resume_train"]:
            self.uepoch = self.config["load_epoch"]
            self.uidx["pretrained"] = self.uepoch * self.validFreq
Exemplo n.º 2
0
 def __init__(self, port, config, device):
     PTServer.__init__(self, port = port, \
                             config = config, \
                             device = device)
     self.config['irank'] = 0
     
     self.max_mb = self.config['n_epochs'] * len(self.data[0])
     print 'max_mb = %d' % self.max_mb
     self.uidx = {}
     self.valid = {}
     self.adj_lr = {}
     self.start_time = None
     self.last = None
     self.last_uidx = 0
     self.validFreq = len(self.data[0])
     self.uepoch = 0
     self.last_uepoch = 0
     
     if self.config['resume_train']:
         self.uepoch = self.config['load_epoch']
         self.uidx['pretrained'] = self.uepoch * self.validFreq
Exemplo n.º 3
0
    def __init__(self, port, config, device):
        PTServer.__init__(self, port = port, \
                                config = config, \
                                device = device)
        self.config['irank'] = 0

        self.max_mb = self.config['n_epochs'] * len(self.data[0])
        print 'max_mb = %d' % self.max_mb
        self.uidx = {}
        self.valid = {}
        self.adj_lr = {}
        self.start_time = None
        self.last = None
        self.last_uidx = 0
        self.validFreq = len(self.data[0])
        self.uepoch = 0
        self.last_uepoch = 0

        if self.config['resume_train']:
            self.uepoch = self.config['load_epoch']
            self.uidx['pretrained'] = self.uepoch * self.validFreq