Compare commits
2 Commits
18df233e6d
...
0b15d28a5b
Author | SHA1 | Date | |
---|---|---|---|
|
0b15d28a5b | ||
|
aafe737020 |
4
.github/workflows/train.yml
vendored
4
.github/workflows/train.yml
vendored
@ -15,12 +15,12 @@ jobs:
|
||||
|
||||
- name: Use Docker action
|
||||
id: train
|
||||
uses: ./
|
||||
uses: ./train-action/
|
||||
with:
|
||||
epochs: 1
|
||||
|
||||
- name: Archive
|
||||
uses: action/upload-artifact@v3
|
||||
uses: actions/upload-artifact@v3
|
||||
with:
|
||||
name: model
|
||||
path: model.keras
|
||||
|
@ -14,4 +14,4 @@ RUN wget https://bendun.students.wmi.amu.edu.pl/stop_times.categories.tsv
|
||||
|
||||
VOLUME /github/workspace/
|
||||
|
||||
ENTRYPOINT ["/ium/train-entrypoint.sh"]
|
||||
ENTRYPOINT ["/ium/train-action/entrypoint.sh"]
|
@ -7,6 +7,6 @@ inputs:
|
||||
default: '2'
|
||||
runs:
|
||||
using: 'docker'
|
||||
image: 'train.dockerfile'
|
||||
image: 'Dockerfile'
|
||||
args:
|
||||
- ${{ inputs.epochs }}
|
Loading…
Reference in New Issue
Block a user