Permalink
Switch branches/tags
Nothing to show
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
21 lines (15 sloc) 616 Bytes
import os
if not os.path.exists('logs'):
os.makedirs('logs')
test_files = [f for f in os.listdir("test_files") if f.endswith('.py')]
test_files.remove('CustomCallback.py')
backends = ['cntk', 'tensorflow']
docker_cmd = "sudo nvidia-docker run -it --rm -v $(pwd)/:/keras --name keras"
for test_file in test_files:
for backend in backends:
statement = docker_cmd + \
" -e KERAS_BACKEND='{}' minimaxir/keras-cntk python3 test_files/{}".format(
backend, test_file)
print(statement)
print("{} + {}".format(test_file, backend))
os.system(statement)