X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pytorch.git;a=blobdiff_plain;f=denoising-ae-field.py;h=2aa3648991583c7681d145d727a7df5b40de5616;hp=8f748d11a3ff219922220b4ed9f14b3eb473a2e1;hb=7b9a3c65d7d1f534de76b49ddabca70c69773390;hpb=72e4bc5d20e153800f19a94e4bfd075adf30e3f3

diff --git a/denoising-ae-field.py b/denoising-ae-field.py
index 8f748d1..2aa3648 100755
--- a/denoising-ae-field.py
+++ b/denoising-ae-field.py
@@ -61,7 +61,7 @@ def train_model(data):
 
 ######################################################################
 
-def save_image(data, data_name, model):
+def save_image(data_name, model, data):
     a = torch.linspace(-1.5, 1.5, 30)
     x = a.view( 1, -1, 1).expand(a.size(0), a.size(0), 1)
     y = a.view(-1,  1, 1).expand(a.size(0), a.size(0), 1)
@@ -104,4 +104,4 @@ for data_source in [ data_zigzag, data_spiral, data_penta ]:
     data, data_name = data_source(1000)
     data = data - data.mean(0)
     model = train_model(data)
-    save_image(data, data_name, model)
+    save_image(data_name, model, data)