예제 #1
0
 def build_model(self, args):
     from disent import models
     return nn.ModuleDict({
         'main': models.build_model(args),
         'adv1': models.build_adversarial(args),
         'adv2': models.build_adversarial(args),
     })
예제 #2
0
 def build_model(self, args):
     from disent import models
     return models.build_model(args)