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