jprdonnelly/MachineLearningNotebooks
1
0
Fork 0
mirror of https://github.com/Azure/MachineLearningNotebooks.git synced 2025-12-20 01:27:06 -05:00
Code Issues Projects Releases Wiki Activity

Update notebooks

Browse Source
This commit is contained in:
Roope Astala
2018-10-12 14:39:33 -04:00
parent 216aa8b6a1
commit a4792d95ac
67 changed files with 6470 additions and 1610 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
training/01.train-hyperparameter-tune-deploy-with-pytorch/pytorch_train.py
View File

@@ -17,7 +17,7 @@ import argparse
from azureml.core.run import Run
# get the Azure ML run object
run = Run.get_submitted_run()
run = Run.get_context()
def load_data(data_dir):
@@ -162,8 +162,8 @@ def main():
parser.add_argument('--data_dir', type=str, help='directory of training data')
parser.add_argument('--num_epochs', type=int, default=25, help='number of epochs to train')
parser.add_argument('--output_dir', type=str, help='output directory')
parser.add_argument('--learning_rate', type=float, help='learning rate')
parser.add_argument('--momentum', type=float, help='momentum')
parser.add_argument('--learning_rate', type=float, default=0.001, help='learning rate')
parser.add_argument('--momentum', type=float, default=0.9, help='momentum')
args = parser.parse_args()
print("data directory is: " + args.data_dir)