From 92fb03c79399e83fe01057f34f82e2acf524e527 Mon Sep 17 00:00:00 2001 From: Bing Xu Date: Tue, 10 Jun 2014 21:06:16 -0400 Subject: [PATCH] new --- tfd_pretrain/pretrain.yaml | 6 +++--- tfd_pretrain/train.yaml | 8 ++++---- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/tfd_pretrain/pretrain.yaml b/tfd_pretrain/pretrain.yaml index f7d4c6b..6a9dbcf 100644 --- a/tfd_pretrain/pretrain.yaml +++ b/tfd_pretrain/pretrain.yaml @@ -3,8 +3,8 @@ which_set: 'unlabeled', scale: True, }, - model: !obj:galatea.adversarial.AdversaryPair { - generator: !obj:galatea.adversarial.Generator { + model: !obj:adversarial.AdversaryPair { + generator: !obj:adversarial.Generator { monitor_ll: 1, mlp: !obj:pylearn2.models.mlp.MLP { layers: [ @@ -82,7 +82,7 @@ # scale: True, # } }, - cost: !obj:galatea.adversarial.AdversaryCost2 { + cost: !obj:adversarial.AdversaryCost2 { scale_grads: 0, #target_scale: 1., discriminator_default_input_include_prob: .5, diff --git a/tfd_pretrain/train.yaml b/tfd_pretrain/train.yaml index a17ff0b..84ca0f7 100644 --- a/tfd_pretrain/train.yaml +++ b/tfd_pretrain/train.yaml @@ -3,10 +3,10 @@ which_set: 'unlabeled', scale: True, }, - model: !obj:galatea.adversarial.AdversaryPair { - generator: !obj:galatea.adversarial.Generator { + model: !obj:adversarial.AdversaryPair { + generator: !obj:adversarial.Generator { monitor_ll: 1, - mlp: !obj:galatea.adversarial.add_layers { + mlp: !obj:adversarial.add_layers { mlp: !obj:pylearn2.models.mlp.MLP { layers: [ !obj:pylearn2.models.mlp.RectifiedLinear { @@ -79,7 +79,7 @@ # scale: True, # } }, - cost: !obj:galatea.adversarial.AdversaryCost2 { + cost: !obj:adversarial.AdversaryCost2 { scale_grads: 0, #target_scale: 1., discriminator_default_input_include_prob: .5,