with open(discriminator_sequence_filename, "r") as f:
try:
discriminator_params = json.load(f)
except Exception as e:
raise Exception(
"could not load {}".format(discriminator_sequence_filename))
else:
config = DiscriminatorParams()
config.a = 0
config.b = 1
config.c = 1
config.weight_std = 0.01
config.weight_initializer = "Normal"
config.use_weightnorm = False
config.nonlinearity = "leaky_relu"
config.optimizer = "adam"
config.learning_rate = 0.0001
config.momentum = 0.5
config.gradient_clipping = 1
config.weight_decay = 0
discriminator = Sequential()
discriminator.add(Linear(None, 128, use_weightnorm=config.use_weightnorm))
discriminator.add(Activation(config.nonlinearity))
# discriminator.add(BatchNormalization(128))
discriminator.add(Linear(None, 128, use_weightnorm=config.use_weightnorm))
discriminator.add(Activation(config.nonlinearity))
# discriminator.add(BatchNormalization(128))
discriminator.add(Linear(None, 1, use_weightnorm=config.use_weightnorm))
discriminator_params = {
if os.path.isfile(discriminator_sequence_filename):
print "loading", discriminator_sequence_filename
with open(discriminator_sequence_filename, "r") as f:
try:
params = json.load(f)
except Exception as e:
raise Exception(
"could not load {}".format(discriminator_sequence_filename))
else:
config = DiscriminatorParams()
config.weight_init_std = 0.001
config.weight_initializer = "Normal"
config.use_weightnorm = False
config.nonlinearity = "elu"
config.optimizer = "Adam"
config.learning_rate = 0.0001
config.momentum = 0.5
config.gradient_clipping = 10
config.weight_decay = 0
config.use_feature_matching = False
config.use_minibatch_discrimination = False
discriminator = Sequential(weight_initializer=config.weight_initializer,
weight_init_std=config.weight_init_std)
discriminator.add(gaussian_noise(std=0.3))
discriminator.add(
Convolution2D(3,
32,
ksize=4,
stride=2,
with open(discriminator_sequence_filename, "r") as f:
try:
params = json.load(f)
chainer.global_config.discriminator = to_object(params["config"])
except Exception as e:
raise Exception(
"could not load {}".format(discriminator_sequence_filename))
else:
config = DiscriminatorParams()
config.clamp_lower = -0.01
config.clamp_upper = 0.01
config.num_critic = 1
config.weight_std = 0.001
config.weight_initializer = "Normal"
config.nonlinearity = "leaky_relu"
config.optimizer = "rmsprop"
config.learning_rate = 0.0001
config.momentum = 0.5
config.gradient_clipping = 10
config.weight_decay = 0
chainer.global_config.discriminator = config
discriminator = Sequential()
discriminator.add(Convolution2D(3, 32, ksize=4, stride=2, pad=1))
discriminator.add(BatchNormalization(32))
discriminator.add(Activation(config.nonlinearity))
discriminator.add(Convolution2D(32, 64, ksize=4, stride=2, pad=1))
discriminator.add(BatchNormalization(64))
discriminator.add(Activation(config.nonlinearity))
discriminator.add(Convolution2D(64, 128, ksize=4, stride=2, pad=1))