def set_net(net, reg_interval): ret = EasyDict() ret.args = EasyDict(func_name = "training.network.{}_GANsformer".format(net[0])) # network options ret.loss_args = EasyDict(func_name = "training.loss.{}_loss".format(net[0])) # loss options ret.opt_args = EasyDict(beta1 = 0.0, beta2 = 0.99, epsilon = 1e-8) # optimizer options ret.reg_interval = reg_interval return ret
def set_net(net, reg_interval): ret = EasyDict() net2name = {"D": "Discriminator", "G": "Generator"} ret.args = EasyDict( func_name=f"training.networks.{net2name[net]}") # network options ret.loss_args = EasyDict( func_name=f"training.loss.{net}_loss") # loss options ret.opt_args = EasyDict(beta1=0.0, beta2=0.99, epsilon=1e-8) # optimizer options ret.reg_interval = reg_interval return ret