Skip to content

Commit

Permalink
- added train main function.
Browse files Browse the repository at this point in the history
- added upload_data main function.
- modified gitignore for azureml.
  • Loading branch information
Seb-Good committed Apr 30, 2019
1 parent 7c4cc30 commit 8a6d1ec
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 4 deletions.
4 changes: 1 addition & 3 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -171,6 +171,4 @@ ENV/
# VS Cods
.vscode
aml_config/*
assets/*
aml_config
assets
assets/*
3 changes: 3 additions & 0 deletions train.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,13 +57,16 @@ def main(args):

# Loop through train dataset batches
for batch in range(steps_per_epoch):

# Run train operation
loss, accuracy, _, _, _, _ = sess.run(
fetches=[graph.loss, graph.accuracy, graph.train_op, graph.update_metrics_op,
graph.train_summary_metrics_op, graph.global_step],
feed_dict={graph.batch_size: args.batch_size, graph.is_training: True,
graph.learning_rate: args.learning_rate, graph.mode_handle: handle_train}
)

# Print performance
if batch % 100 == 0:
print('Loss: {}, Accuracy: {}'.format(loss, accuracy))

Expand Down
2 changes: 1 addition & 1 deletion upload_data.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,4 +44,4 @@ def get_parser():
arguments = get_parser().parse_args()

# Run main function
main(args=arguments)
main(args=arguments)

0 comments on commit 8a6d1ec

Please sign in to comment.