File size: 531 Bytes
4c62147 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import torchtask
import func, data, model, criterion, trainer
def add_parser_arguments(parser):
torchtask.proxy_template.add_parser_arguments(parser)
data.add_parser_arguments(parser)
model.add_parser_arguments(parser)
criterion.add_parser_arguments(parser)
trainer.add_parser_arguments(parser)
class HarmonizerProxy(torchtask.proxy_template.TaskProxy):
NAME = 'harmonizer'
def __init__(self, args):
super(HarmonizerProxy, self).__init__(args, func, data, model, criterion, trainer)
|