Upload 3 files

atlas3_medformer/config.txt

@@ -0,0 +1,89 @@
+dataset: abdomenatlas
+model: medformer
+dimension: 3d
+pretrain: False
+amp: False
+torch_compile: False
+batch_size: 1
+resume: True
+load: /home/psalvador/JHU/foundational/MedFormer/exp/abdomenatlas/atlas_3_medformer/fold_0_latest.pth
+cp_path: ./exp/
+log_path: ./log/abdomenatlas/
+unique_name: atlas_3_medformer
+use_k_fold: False
+all_train: False
+crop_on_tumor: True
+multi_ch_tumor: False
+multi_ch_tumor_data_root: /home/psalvador/data_multi_ch/
+multi_ch_tumor_classes: 61
+debug_val: False
+workers_overwrite: 4
+load_augmented: True
+save_destination: /home/psalvador/aug_medformer/atlas_3.0_medformer_npy_augmented/
+save_augmented: False
+data_root: /home/psalvador/data/medformer_atlas3/
+id_list: None
+world_size: 4
+rank: 0
+dist_url: tcp://127.0.0.1:12345
+dist_backend: nccl
+gpu: 0,1,2,3
+UFO_root: /fastwork/psalvador/JHU/data/UFO/
+reports: /fastwork/psalvador/JHU/data/UFOLLMOutputLarge.csv
+classes: 42
+modality: CT
+arch: medformer
+in_chan: 1
+base_chan: 32
+conv_block: BasicBlock
+down_scale: [[2, 2, 2], [2, 2, 2], [2, 2, 2], [2, 2, 2]]
+kernel_size: [[3, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3]]
+chan_num: [64, 128, 256, 320, 256, 128, 64, 32]
+norm: in
+act: relu
+map_size: [3, 3, 3]
+conv_num: [2, 0, 0, 0, 0, 0, 2, 2]
+trans_num: [0, 2, 4, 6, 4, 2, 0, 0]
+num_heads: [1, 4, 8, 10, 8, 4, 1, 1]
+expansion: 4
+fusion_depth: 2
+fusion_dim: 320
+fusion_heads: 10
+attn_drop: 0
+proj_drop: 0
+proj_type: depthwise
+epochs: 300
+training_size: [128, 128, 128]
+start_epoch: 0
+num_workers: 4
+aug_device: cpu
+aux_loss: True
+aux_weight: [0.5, 0.5]
+split_seed: 0
+k_fold: 10
+optimizer: adamw
+base_lr: 0.0006
+betas: [0.9, 0.999]
+weight_decay: 0.05
+weight: [0.5, 1, 1, 1, 1, 1, 1, 1, 1, 1]
+rlt: 1
+scale: [0, 0, 0]
+rotate: [30, 30, 30]
+translate: [0, 0, 0]
+affine_pad_size: [40, 40, 40]
+gaussian_noise_std: 0.02
+print_freq: 5
+iter_per_epoch: 1000
+ema: True
+ema_alpha: 0.99
+val_freq: 20000
+sliding_window: True
+window_size: [128, 128, 128]
+proc_idx: 0
+port: 10000
+multiprocessing_distributed: True
+reproduce_seed: None
+distributed: True
+ngpus_per_node: 4
+workers: 1
+cp_dir: ./exp//abdomenatlas/atlas_3_medformer

atlas3_medformer/fold_0_latest.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a8930377276aa0cc5f19f6ba95978aa7dda4fccbc03b25615f6cfccb9fb5a17b
+size 606822769