LVM-Med / dataloader /yaml_data /fgadr_endtoend_R50_frozen_1_fcn.yml
duynhm's picture
Initial commit
be2715b
raw
history blame contribute delete
No virus
449 Bytes
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'