diff --git a/src/electrai/configs/MP/config_resunet.yaml b/src/electrai/configs/MP/config_resunet.yaml
index 7fef6ff..627a9d6 100644
--- a/src/electrai/configs/MP/config_resunet.yaml
+++ b/src/electrai/configs/MP/config_resunet.yaml
@@ -5,7 +5,7 @@ data:
   split_file: null #/scratch/gpfs/ROSENGROUP/common/globus_share_OA/mp/dataset_1/split.json
   precision: f32
   batch_size: 1
-  train_workers: 8
+  train_workers: 4
   val_workers: 2
   pin_memory: false
   val_frac: 0.005
@@ -20,11 +20,11 @@ model:
   _target_: electrai.model.resunet.ResUNet3D
   in_channels: 1
   out_channels: 1
-  n_channels: 32
+  n_channels: 64
   n_residual_blocks: 1
   kernel_size: 5
   depth: 2
-  use_checkpoint: False
+  use_checkpoint: True
 
 # Training parameters
 precision: 32