Skip to content

Commit

Permalink
Create tensorflow_cnn.yaml
Browse files Browse the repository at this point in the history
  • Loading branch information
nily-dti committed Dec 10, 2019
1 parent 58de94a commit c671e57
Showing 1 changed file with 27 additions and 0 deletions.
27 changes: 27 additions & 0 deletions examples/fashion_mnist/tensorflow_cnn.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
- experiment: tensorflow_cnn
description: Using Tracker to do good science!
operations:
train:
parameters:
batch-size:
value: 64
test-batch-size:
value: 1000
epochs:
value: 1
lr:
description: Learning rate
value: 0.01
lr_decay:
description: Learning rate decay
value: 0.0
momentum:
value: 0.5
seed:
description: Preset number (or list of numbers) to set for pseudo-random number generators
value: 42
environments:
- name: tf
type: docker
image: tensorflow/tensorflow:2.0.0-py3
executable: src/train.py

0 comments on commit c671e57

Please sign in to comment.