ResNet50_Classifier/config.yml at main · mehmetharmanli/ResNet50_Classifier · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
model_config:
  model: MobileNetV3Small #ResNet50, MobileNetV3Small
  initial_filter_size: [32]
  kernel: [3]
  dropout_rate: [0.5]
  input_size: [800, 800] #h, w
  channel: 1
  num_class: 2
  attention: False
train_config:
  loss: w_ce
  accuracy: dice_score
  optimizer: Adam
  lr_rate: [0.00005]
  weight_decay: [0.0002]
  batch_size: [4]
  epochs: 10
  num_workers: 20
  use_cuda: True
  seed: 1636
dataset_config:
  train_path: '/home/ocaki13/projects/ultrasound/processed_data/ultrasoundSegmentationDatasetv2_resized_won_cls/fold1/train/'
  val_path: '/home/ocaki13/projects/ultrasound/processed_data/ultrasoundSegmentationDatasetv2_resized_won_cls//fold1/val/'
  test_path: '/home/ocaki13/projects/ultrasound/processed_data/ultrasoundSegmentationDatasetv2_resized_won_cls//fold1/test/'
  augmentation: True
  aug_rate: [0]
  save_dir: results4
  class_names: ['control','RD']
resume:
  flag: False
  path: exp3/models/last_epoch.pt
  epoch: 10