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