You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
54 lines
902 B
54 lines
902 B
# One-Prompt Medical Image Segmentation - Development Configuration
|
|
|
|
project:
|
|
name: "one-prompt-segmentation"
|
|
version: "1.0.0"
|
|
description: "开发环境配置"
|
|
mode: "development"
|
|
|
|
# 数据配置
|
|
data:
|
|
dataset: "polyp"
|
|
data_path: "./data/TestDataset"
|
|
train_ratio: 0.8
|
|
batch_size: 2
|
|
num_workers: 2
|
|
|
|
# 模型配置
|
|
model:
|
|
net: "oneprompt"
|
|
baseline: "unet"
|
|
mod: "one_adpt"
|
|
image_size: 256
|
|
out_size: 256
|
|
patch_size: 16
|
|
dim: 256
|
|
depth: 1
|
|
heads: 16
|
|
mlp_dim: 1024
|
|
|
|
# 训练配置 (开发环境使用较小的配置)
|
|
training:
|
|
epochs: 10
|
|
learning_rate: 0.0001
|
|
optimizer: "adam"
|
|
weight_decay: 0.0
|
|
scheduler:
|
|
name: "step"
|
|
step_size: 5
|
|
gamma: 0.5
|
|
gradient_clip: 1.0
|
|
|
|
# 验证配置
|
|
validation:
|
|
val_freq: 2
|
|
vis_freq: 10
|
|
|
|
# 日志配置
|
|
logging:
|
|
log_dir: "logs/dev"
|
|
tensorboard: true
|
|
save_best: true
|
|
checkpoint_freq: 5
|
|
log_level: "DEBUG"
|