From 8ea0e3c5cc303718a8b508b656f7aa9e64ea3070 Mon Sep 17 00:00:00 2001
From: Francois Fleuret <francois@fleuret.org>
Date: Wed, 27 Jul 2022 11:18:06 +0200
Subject: [PATCH] Update.

---
 main.py                  |   6 +++---
 result_picoclvr_0007.png | Bin 623 -> 0 bytes
 result_picoclvr_0009.png | Bin 0 -> 872 bytes
 3 files changed, 3 insertions(+), 3 deletions(-)
 delete mode 100644 result_picoclvr_0007.png
 create mode 100644 result_picoclvr_0009.png

diff --git a/main.py b/main.py
index 6592204..b2adf98 100755
--- a/main.py
+++ b/main.py
@@ -111,7 +111,7 @@ for n in vars(args):
 ######################################################################
 
 def autoregression(
-        model,
+        model, batch_size,
         nb_samples, nb_tokens_to_generate, starting_input = None,
         device = torch.device('cpu')
 ):
@@ -126,7 +126,7 @@ def autoregression(
         first = starting_input.size(1)
         results = torch.cat((starting_input, results), 1)
 
-    for input in results.split(args.batch_size):
+    for input in results.split(batch_size):
         for s in tqdm.tqdm(range(first, input.size(1)), desc = 'synth'):
             output = model(input)
             logits = output[:, s]
@@ -386,7 +386,7 @@ class TaskMNIST(Task):
         return 256
 
     def produce_results(self, n_epoch, model, nb_samples = 64):
-        results = autoregression(model, nb_samples, 28 * 28, device = self.device)
+        results = autoregression(model, self.batch_size, nb_samples, 28 * 28, device = self.device)
         image_name = f'result_mnist_{n_epoch:04d}.png'
         torchvision.utils.save_image(1 - results.reshape(-1, 1, 28, 28) / 255.,
                                      image_name, nrow = 16, pad_value = 0.8)
diff --git a/result_picoclvr_0007.png b/result_picoclvr_0007.png
deleted file mode 100644
index 7baee570f0d73ac7bd89c0fbc4accc3dd2cafabc..0000000000000000000000000000000000000000
GIT binary patch
literal 0
HcmV?d00001

literal 623
zcmV-#0+9WQP)<h;3K|Lk000e1NJLTq002?|001Hg0ssI2Lu?rh0006uNkl<Zc%1E7
zTXuvX3{85vBi&zhf6kidho~Xk#OPq>^k{x&ByR}Zh&)Xb-H*OL9l&Skd;aOVPIH)<
z(los=mxzd7FT`c0ls0bPQ%XckluX@4W%<S2O1CFUipxwOW+5}{nybx&^0nxLza`BH
zj4U5j&Kq};nMZ&}F{-s@n=<E-p#>iFh?#c!(FS-d=CyR7=b%>ixxRd+VMnGi!bHT(
zwebUc+l_24QL6yTSVB_BM_^`@1`(yTScKn}M_P)-bv*huB>i4;lAm7BKO=r{`$lBr
zHW69K{9m{g;8Ts;!E?AVElxw;k49)8o?k+yY%^LDj(Lg9LIK$y?P5TDTJ+xb+Xin<
zGRqW*;a1i5i2i6-MY%x1<{@yT*{QLI%%}-HU^;V7V7jTIXJOpl3(B-vHpVQ+XO86{
zP+bfH5dod=VZxOy3;DtAJ5k2%zVBd}wMAEnqY>TfqROkiyS18t`syVJlh?hpM2j4q
zia`RB#p&|Kg!;M;cTHQ1REZUj3f^7!@V>(Y+e4&vnPu3H<u&c<x?|7-argVRP$)Pv
z?%S!jDLL(yjLI%@e--rlmbEbWs_05gutn(TVrx_P>!fP<L{txM--+=1Hgj&jPaq_`
z>#pB?-$(Tb*YazvHrO^nq=9Eb<hXWZ;vlCaEt@#))1hBPao0h~o9=Bt?KU?pyD(k~
zG`rq$%<8mDQTXx6zA7-pPz&Moe8o2{=m!bxw7~i!+=JUUqF+U9MA`T}_>lkr002ov
JPDHLkV1lCSGkX94

diff --git a/result_picoclvr_0009.png b/result_picoclvr_0009.png
new file mode 100644
index 0000000000000000000000000000000000000000..18dad2782bd2990f67065a4b48c3cf6aa1fbe6d5
GIT binary patch
literal 872
zcmV-u1DE`XP)<h;3K|Lk000e1NJLTq005Ez0024&0ssI2)K-^T0009oNkl<Zc%1EA
zS(4)*427w_QRaSAeZS;P%!lK60j!-R*8AzIwuP355Jq<5^Z5*p+3=^g0S*p+mOqcj
zgCzEPz1~tWhLp%sI4onIrEnToslOp$&0|=oEQN4wk=kWjj3ligg!mgnN|7#zMA-%O
zELrZ>F$Iq`8l0%CuEk$Kct;y893g~|{-qE?q6!+&4ElN0kG`2<>KcttR2b8G3G4;R
zwT%=7GA?OmsE!fx^BB04P3k<b?@K#V*^@0-Cr6n!kZn6KRKwX`x#OcNzn+^xUyR@}
z*(lIvR;nCoD9|RKRt{_UbK8_~+NOllW&!ivfdXy5g~?#PJ5Zp_w=fpCPKM|mfEnSn
zZAyqc0DYzEN9HJ9nYiRszGXBB*=Bdr&xf$CV#Kvew@MUow7pgN-a`Q@(*Q?XN?Q6R
z#CU6r&6eco=YUHq4Z5>cCmKB3u5};%yiJng+G$`kP`vVJvtvRs89S#n+)hI+JlP;6
zDU;_m!MZ4!5W}Z6YvG9{zf5QY&6~d7HAFF(eNIVkQ}4(?J{%8!xTT+%O*5T(MuB_{
zZg|=>-kmy)nEwkLS^nA<7iGoX0hkiQXPcvh$vXfzDg1W|v-qWxZG&&<zi;EZ(IyN1
z*u2$Y``Wx2Hrn84<hqCA{ZcnnPNe#9iO-zVB6%g(@xy=*TCK*Z%d(H3{3qH>h$*Hi
zFD}ipnxLo^!}~C5y_7&sQ^}e<0ygswo*3G^4<muL%L1mK1?)As4c`^>&^B*5z_GUB
z<F?vj0*o_44TiDh0LR*fw4IZc%_5r>+bnER<5nFV54X&6q8h)tMvaiS*e<O)(6lw)
z5lhrD;@AASQ&@n#1284L1284}cL2cS`B_fo=j#CEz}Kb*Wu^T|m9<8IHbFC<DQE$?
zG1NGyxPPG(plzt=U&C~QcZtJ=((l=lOf<vf_#WDB1*eUU0bapy+UOWyIldchv_w(r
z0qK86r15hyfN!0@eL9*)OO&h!#Ir~LGa`^S*ifjyRlSOs%P5DB<DAD-ZNHf>kI4K6
yX-js^Vy%E4$LU{qBoRZ##5({};%-}#FTsC_>{b4z$`0HB0000<MNUMnLSTYvtE4pm

literal 0
HcmV?d00001

-- 
2.20.1