finalize upscaler conf

configs/stable-diffusion/txt2img-upscale-clip-encoder-f16-1024.yaml

@@ -66,7 +66,7 @@ model:
         image_size: 64    # not really needed
         in_channels: 20
         out_channels: 16
-        model_channels: 32 # TODO: more
+        model_channels: 192
         attention_resolutions: [ 4, 2, 1 ]
         num_res_blocks: 2
         channel_mult: [ 1, 2, 4, 4 ]
@@ -105,8 +105,8 @@ data:
   target: ldm.data.laion.WebDataModuleFromConfig
   params:
     tar_base: "pipe:aws s3 cp s3://s-datasets/laion-high-resolution/"
-    batch_size: 4
-    num_workers: 1
+    batch_size: 8
+    num_workers: 4
     train:
       shards: '{00000..17279}.tar -'
       shuffle: 10000
@@ -147,7 +147,7 @@ lightning:
     image_logger:
       target: main.ImageLogger
       params:
-        batch_frequency: 10
+        batch_frequency: 5000
         max_images: 4
         increase_log_steps: False
         log_first_step: False
@@ -164,4 +164,4 @@ lightning:
     benchmark: True
     val_check_interval: 5000000  # really sorry
     num_sanity_val_steps: 0
-    accumulate_grad_batches: 1
+    accumulate_grad_batches: 4