martin-fabbri
/
tpu-workflow
mirror of https://github.com/martin-fabbri/tpu-workflow.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

	
32

	
33

	
34

	
35

	
36

	
37

	
38

	
39

	
40

	
            dataset:
  image_size: 331
  # do not end the gcs_path with a backslash
  gcs_path: gs://kaggle-data-tpu/images
  gcs_path_remote_alias: gcs_images
  classes:
    - daisy
    - dandelion
    - roses
    - sunflowers
    - tulips
  classes_path: data/domain/classes.json

split:
  train_split_path: data/interim/train_split.json
  val_split_path: data/interim/val_split.json
  images_count_path: data/interim/images_count.json
  validation_split: 0.19

train:
  nb_validation_samples: 800
  epochs: 12
  batch_size: 16
  lr:
    start: 0.00001
    max: 0.00005
    min: 0.00001
    rampup_epochs: 5
    sustain_epochs: 0
    exp_decay: 0.8
  saved_model_path: models/xception_model.h5
  train_plot_path: plots/train_accuracy_plot.json
evaluate:
  evaluate_metrics_path: metrics/evaluate_metrics.json
checkpoint:
  dir_path: checkpoints
  prefix: cp.ckpt
  index: cp.ckpt.index
  volume: cp.ckpt.data-00000-of-00001