Dean
/
dvc-streamlit-example
mirror of https://github.com/sicara/dvc-streamlit-example.git


  
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

	
            root_dir: ./data

data:
  download:
    subdir: raw
    url: https://storage.googleapis.com/mledu-datasets/cats_and_dogs_filtered.zip
    zip_file: cats_and_dogs_filtered.zip
  dataset:
    subdir: dataset
    val_test_split:
      val: 0.7
      test: 0.3

train:
  batch_size: 32
  img_size: [160, 160]
  learning_rate: 0.0001
  subdir: train
  epochs:
    frozen: 10
    unfrozen: 15
  fine_tune_at: 100

model:
  # Choose among available models in tf.keras.applications
  # See https://www.tensorflow.org/api_docs/python/tf/keras/applications
  backbone: tensorflow.keras.applications.ResNet50
  preprocess_input: tensorflow.keras.applications.resnet50.preprocess_input

evaluate:
  subdir: evaluation