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