update model.py

model.py

@@ -55,9 +55,9 @@ class SelectiveAttention:
 
 
 class AlignDRAW(nn.Module):
-    def __init__(self, dimY=25323, dimLangRNN=128, dimAlign=512, imgSize=32,
+    def __init__(self, dimY=25323, dimLangRNN=128, dimAlign=512, imgSize=512,
                  dimReadAttent=9, dimWriteAttent=9, dimRNNEnc=550, dimRNNDec=550,
-                 dimZ=275, runSteps=32):
+                 dimZ=275, runSteps=69):
         super().__init__()
         self.C = 3
         self.H = self.W = imgSize