diff --git a/.github/workflows/sm_train.yml b/.github/workflows/sm_train.yml index 9d5be2b..a67dff5 100644 --- a/.github/workflows/sm_train.yml +++ b/.github/workflows/sm_train.yml @@ -17,6 +17,8 @@ jobs: - name: Setup Python environment uses: actions/setup-python@v1.1.1 + env: + ACTIONS_ALLOW_UNSECURE_COMMANDS: 'true' - name: Install requirements run: pip install --requirement requirements.txt @@ -33,4 +35,4 @@ jobs: run: | python sm_train.py - \ No newline at end of file + diff --git a/sm_train.py b/sm_train.py index ccc2a77..8cf2ef2 100644 --- a/sm_train.py +++ b/sm_train.py @@ -20,7 +20,7 @@ {'Name': 'train:epoch', 'Regex': 'Train_epoch=(.*?);'} ] - + estimator = PyTorch(entry_point='mnist.py', source_dir='code', role=role,