Skip to content

Instantly share code, notes, and snippets.

@sengstacken

sengstacken/test_local.py

Created August 17, 2022 17:55
Show Gist options
  • Save sengstacken/0cf7e1208e4266e966bff26c7c9034c5 to your computer and use it in GitHub Desktop.
Save sengstacken/0cf7e1208e4266e966bff26c7c9034c5 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
test_local.py
# from - https://gitlab.com/juliensimon/aim410/-/blob/master/aim410.ipynb
# Number of GPUs on this machine
%env SM_NUM_GPUS=0
# Where to save the model
%env SM_MODEL_DIR=/tmp/model
# Where the training data is
%env SM_CHANNEL_TRAINING=data
# Where the validation data is
%env SM_CHANNEL_VALIDATION=data
!/bin/rm -rf $SM_MODEL_DIR
!python mnist_keras_tf.py --epochs 1
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment