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
