name: Train
description: 'Train model'
inputs:
epochs:
description: 'Number of epochs'
required: true
default: '2'
runs:
using: 'docker'
image: 'Dockerfile'
args:
- ${{ inputs.epochs }}
