File size: 449 Bytes
be2715b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
base:
is_R50: 1
is_SAMVIT: 0
dataset_name: fgadr
gpu_id: '3'
original_checkpoint: 'lvm-med-resnet'
best_valid_model_checkpoint: './working_dir/checkpoint/'
frozen_eval: 1
model: '1-fcn'
test_mode: 'best_valid'
valid_rate: 0.2
train:
train_batch_size: 32
valid_batch_size: 1
test_batch_size: 1
learning_rate: 0.0005
num_epochs: 20
linear_eval: 0
solver: 'adam'
dataloader:
data_path: './dataset_demo/FGADR'
|