Esempi in Python per Operator.Operator

Esempio n. 1

File: main.py Progetto: peterhan91/what

def main(config):
    config.device = torch.device(
        'cuda:{}'.format(config.gpu) if torch.cuda.is_available() else 'cpu')

    # load data_loader
    data_loader = get_dataloader(config)
    check_point = Checkpoint(config)
    operator = Operator(config, check_point)

    if config.is_train:
        operator.train(data_loader)
    else:
        operator.test(data_loader)

Esempio n. 2

File: main.py Progetto: into-the-unk/prnn

def main(config):
    config.device = torch.device(
        'cuda:{}'.format(config.gpu) if torch.cuda.is_available() else 'cpu')

    # load data_loader
    check_point = Checkpoint(config)
    operator = Operator(config, check_point)

    if config.is_train:
        midi_data = MidiData(config)
        operator.train(midi_data)
    else:
        operator.test()

Esempio n. 3

 def addOp(self, id):
     operator = Operator(id)
     operator.status = 'available'
     self.operators.append(operator)