new

zynz · Jun 11, 2014 · 92fb03c · 92fb03c
1 parent 39c1777
commit 92fb03c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git 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
@@ -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,