test_estimator.yml 537 Bytes
Newer Older
Corey Lynch's avatar
Corey Lynch committed
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
use_tpu: False
training_strategy: 'mvtcn'
loss_strategy: 'triplet_semihard'

learning:
  max_step: 2
  optimizer: 'adam'

embedding_size: 8

data:
  embed_batch_size: 12
  batch_size: 12
  examples_per_sequence: 12
  num_views: 2
  num_parallel_calls: 1
  sequence_prefetch_size: 1
  batch_prefetch_size: 1

logging:
  summary:
    image_summaries: false
    save_summaries_steps: 100
    flush_secs: 600
    save_summaries_secs: 60
  checkpoint:
    num_to_keep: 0 # Keep all checkpoints.
    save_checkpoints_steps: 1000
    secs: 1800