def __init__(self, root_path):
        """Initialize paths and loggers of the model."""
        # Storage director of the model and its snapshots
        self.root_path = root_path
        self.model_path = os.path.join(self.root_path, self.Model_Filename)

        utils.remove_if_exists(self.model_path)

        # Set logging output...
        for name in ["neon.util.persist"]:
            dslogger = logging.getLogger(name)
            dslogger.setLevel(40)

        print 'Epochs: %d Batch-Size: %d' % (self.max_epochs, self.batch_size)
Exemple #2
0
    def __init__(self, root_path, lrate=0.01, batch_size=30, max_epochs=10):
        """Initialize paths and loggers of the model."""
        # Storage director of the model and its snapshots
        self.root_path = root_path
        self.model_name = os.path.basename(self.root_path)
        self.model_path = os.path.join(self.root_path, self.Model_Filename)
        utils.remove_if_exists(self.model_path)

        # Training settings
        self.lrate = lrate
        self.batch_size = batch_size
        self.max_epochs = max_epochs

        # Set logging output...
        for name in ["neon.util.persist"]:
            dslogger = logging.getLogger(name)
            dslogger.setLevel(40)

        print "[%s] Epochs: %d Batch-Size: %d" % (self.model_name, self.max_epochs, self.batch_size)
    def __init__(self, root_path, lrate=0.01, batch_size=30, max_epochs=10):
        """Initialize paths and loggers of the model."""
        # Storage director of the model and its snapshots
        self.root_path = root_path
        self.model_name = os.path.basename(self.root_path)
        self.model_path = os.path.join(self.root_path, self.Model_Filename)
        utils.remove_if_exists(self.model_path)
        
        # Training settings
        self.lrate = lrate
        self.batch_size = batch_size
        self.max_epochs = max_epochs

        # Set logging output...
        for name in ["neon.util.persist"]:
            dslogger = logging.getLogger(name)
            dslogger.setLevel(40)

        print "[%s] Epochs: %d Batch-Size: %d" % (self.model_name, self.max_epochs, self.batch_size)