fix cspca inference

config/config_cspca_train_2.yaml

@@ -0,0 +1,17 @@
+data_root: /sc-projects/sc-proj-cc06-ag-ki-radiologie/pirad_model_test_PICAI/registered/t2_hist_matched/
+dataset_json: /sc-scratch/sc-scratch-cc06-ag-ki-radiologie/prostate_foundation/WSAttention-Prostate/dataset/PICAI_cspca.json
+num_classes: !!int 4
+mil_mode: att_trans
+tile_count: !!int 24
+tile_size: !!int 64
+depth: !!int 3
+use_heatmap: !!bool True
+workers: !!int 6
+checkpoint_pirads: /sc-scratch/sc-scratch-cc06-ag-ki-radiologie/prostate_foundation/WSAttention-Prostate/logs/pirads_training/model_70.pt
+epochs: !!int 80
+batch_size: !!int 8
+optim_lr: !!float 2e-5
+
+
+
+

job_scripts/train_cspca_2.sh

@@ -0,0 +1,19 @@
+#!/bin/bash
+#SBATCH --job-name=cspca_training_70       # Specify job name
+#SBATCH --partition=gpu               # Specify partition name
+#SBATCH --mem=128G   
+#SBATCH --gres=gpu:1             
+#SBATCH --time=48:00:00               # Set a limit on the total run time
+#SBATCH --output=/sc-scratch/sc-scratch-cc06-ag-ki-radiologie/prostate_foundation/WSAttention-Prostate/logs/%x/log.o%j      # File name for standard output
+#SBATCH --error=/sc-scratch/sc-scratch-cc06-ag-ki-radiologie/prostate_foundation/WSAttention-Prostate/logs/%x/log.e%j        # File name for standard error output
+#SBATCH --mail-user=anirudh.balaraman@charite.de
+#SBATCH --mail-type=END,FAIL
+
+
+source /etc/profile.d/conda.sh
+conda activate foundation
+
+RUNDIR="/sc-scratch/sc-scratch-cc06-ag-ki-radiologie/prostate_foundation/WSAttention-Prostate"
+
+
+srun python -u $RUNDIR/run_cspca.py --mode train --config $RUNDIR/config/config_cspca_train_2.yaml

run_cspca.py

@@ -51,11 +51,11 @@ def main_worker(args):
                 cspca_model, train_loader, optimizer, epoch=epoch, args=args
             )
             logging.info(
-                f"STATE {st} EPOCH {epoch} TRAIN loss: {train_loss:.4f} AUC: {train_auc:.4f}"
+                f"EPOCH {epoch} TRAIN loss: {train_loss:.4f} AUC: {train_auc:.4f}"
             )
             val_metric = val_epoch(cspca_model, valid_loader, epoch=epoch, args=args)
             logging.info(
-                f"STATE {st} EPOCH {epoch} VAL loss: {val_metric['loss']:.4f} AUC: {val_metric['auc']:.4f}"
+                f"EPOCH {epoch} VAL loss: {val_metric['loss']:.4f} AUC: {val_metric['auc']:.4f}"
             )
             if val_metric["loss"] < old_loss:
                 old_loss = val_metric["loss"]