Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Train and Test Improvements #3

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Conversation

jajsmith
Copy link
Collaborator

Train & Test Script Improvements

  1. Cleanup

A large chunk of code in the train-and-test.py is not being referenced or used so I've removed it to pare down the script and make it easier to work with:

  • Removed the StreamTee class which replicates the unix command tee
  • Removed the train() function which has been replaced by the improved train_network()
  1. Coming in other PRs to this branch.

I'll make smaller PRs to this train-test-improvements branch so the changes can be broken down easier.

Steps to Test:

Run the script using the train_config.ini configuration:

python train_and_test.py --train_gpu 0 --test_gpu 1 --run_inference True --config train_config.ini

Confirm that training proceeds as expected.

@jajsmith jajsmith self-assigned this May 11, 2018
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant