Count day of the year in inputs.

climax/main/downscale_train.py

@@ -126,7 +126,8 @@ if __name__ == '__main__':
             outputs = 3
 
         # instanciate network
-        net = NET(state_file, len(Era5_ds.data_vars), outputs, filters=FILTERS)
+        inputs = len(Era5_ds.data_vars) + 2 if DOY else len(Era5_ds.data_vars)
+        net = NET(state_file, inputs, outputs, filters=FILTERS)
 
         	# initialize optimizer
         optimizer = torch.optim.Adam(net.parameters(), lr=LR,