From 9977fae472b4d8dd102d3a0256d7dbbe10e3d11c Mon Sep 17 00:00:00 2001
From: Benjamin Morris
 <benjamin.morris@prd-aics-dcp-037.corp.alleninstitute.org>
Date: Mon, 1 Jul 2024 11:42:00 -0700
Subject: [PATCH] more configs

---
 configs/experiment/contrastive/vicreg.yaml |  2 --
 configs/model/contrastive/vicreg.yaml      | 10 ++++------
 2 files changed, 4 insertions(+), 8 deletions(-)

diff --git a/configs/experiment/contrastive/vicreg.yaml b/configs/experiment/contrastive/vicreg.yaml
index 4b9815c4..a49a51c3 100644
--- a/configs/experiment/contrastive/vicreg.yaml
+++ b/configs/experiment/contrastive/vicreg.yaml
@@ -23,5 +23,3 @@ trainer:
   max_epochs: 1000
   check_val_every_n_epoch: 10
   devices: [0]
-
-
diff --git a/configs/model/contrastive/vicreg.yaml b/configs/model/contrastive/vicreg.yaml
index 9218227e..47c1a550 100644
--- a/configs/model/contrastive/vicreg.yaml
+++ b/configs/model/contrastive/vicreg.yaml
@@ -1,6 +1,5 @@
 _target_: cyto_dl.models.contrastive.Contrastive
 
-
 anchor_key: image
 positive_key: image_aug
 target_key: target
@@ -11,17 +10,16 @@ backbone:
   _target_: monai.networks.nets.Regressor
   in_shape: [1, 28, 28] # CYX, should be CZYX for 3D
   out_shape: ${model._aux.latent_dim}
-  channels: [2, 4, 6, 8]
-  strides: [2, 2, 2, 2]
+  channels: [2, 4, 8]
+  strides: [2, 2, 2]
   kernel_size: 3
 
-
 task_head:
   _target_: cyto_dl.nn.head.vic_reg.VICRegHead
-  loss: 
+  loss:
     _target_: cyto_dl.nn.losses.vic_reg.VICRegLoss
     num_features: ${model._aux.latent_dim}
-  dimensions: 
+  dimensions:
     - ${model._aux.latent_dim}
     - 256
     - 256