allennlp_tutorial_estimator2.py

allennlp_tutorial_estimator2.py

estimator = PyTorch(entry_point="tuning.py",
                    source_dir="../../allennlp-sagemaker-tuning",
                    dependencies=[from_root("example"), from_root(".venv")],
                    role=role,
                    framework_version="1.0.0",
                    train_instance_count=1,
                    train_instance_type="ml.p2.8xlarge",
                    hyperparameters={
                        "train-file-name": os.path.basename(s3_paths[0]),
                        "validation": os.path.basename(s3_paths[1]),
                        "epochs": 10
                    })