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
|
- name: Use Docker action
|
||||||
id: train
|
id: train
|
||||||
uses: ./
|
uses: ./train-action/
|
||||||
with:
|
with:
|
||||||
epochs: 1
|
epochs: 1
|
||||||
|
|
||||||
- name: Archive
|
- name: Archive
|
||||||
uses: action/upload-artifact@v3
|
uses: actions/upload-artifact@v3
|
||||||
with:
|
with:
|
||||||
name: model
|
name: model
|
||||||
path: model.keras
|
path: model.keras
|
||||||
|
@ -14,4 +14,4 @@ RUN wget https://bendun.students.wmi.amu.edu.pl/stop_times.categories.tsv
|
|||||||
|
|
||||||
VOLUME /github/workspace/
|
VOLUME /github/workspace/
|
||||||
|
|
||||||
ENTRYPOINT ["/ium/train-entrypoint.sh"]
|
ENTRYPOINT ["/ium/train-action/entrypoint.sh"]
|
@ -7,6 +7,6 @@ inputs:
|
|||||||
default: '2'
|
default: '2'
|
||||||
runs:
|
runs:
|
||||||
using: 'docker'
|
using: 'docker'
|
||||||
image: 'train.dockerfile'
|
image: 'Dockerfile'
|
||||||
args:
|
args:
|
||||||
- ${{ inputs.epochs }}
|
- ${{ inputs.epochs }}
|
Loading…
Reference in New Issue
Block a user