Skip to content

Training Workflow

Training Workflow #77

name: Training Workflow
on:
workflow_dispatch:
inputs:
script_content:
description: 'Content of train.py'
required: true
type: string # Explicitly specify the type
jobs:
train:
runs-on: [self-hosted, gpumode-nvidia-arc]
steps:
- name: Install dependencies
run: |
pip install numpy torch
- name: Create and run training script
run: |
echo "${{ inputs.script_content }}" > train.py
cat train.py # Debug: print the content
python train.py > training.log 2>&1
- name: Upload logs
uses: actions/upload-artifact@v3
if: always() # Upload logs whether the job succeeds or fails
with:
name: training-logs
path: training.log