open_parameters_cifar_10.py

with open(param_path, 'r') as params:
    hyperParams = json.load(params)
print("Hyper parameters: " + str(hyperParams))

lr = float(hyperParams.get('lr', '0.001'))
batch_size = int(hyperParams.get('batch_size', '64'))
epochs = int(hyperParams.get('epochs', '100'))
gpu_count = int(hyperParams.get('gpu_count', '1'))
num_classes = 10

# Read input data config passed by SageMaker
with open(data_path, 'r') as params:
    inputParams = json.load(params)
print("Input parameters: " + str(inputParams))