Skip to content

Add dockerfile

Add dockerfile #2

Workflow file for this run

name: ci docker
on:
push:
branches: [ main ]
pull_request:
branches: [ main ]
jobs:
build:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- uses: docker/setup-qemu-action@v3
- uses: docker/setup-buildx-action@v3
- name: Build image
run: docker build . -t dmma
- name: Test train
run: |
docker run -v $(pwd):/mnt \
dmma \
--config=/mnt/configs/config.py \
--infile=/mnt/data/x_train.csv \
--outfile=/mnt/params.pkl \
--config.training.n_iter=1
- name: Test predict
run: |
docker run -v $(pwd):/mnt \
dmma \
--mode=predict \
--config=/mnt/configs/config.py \
--checkpoint=/mnt/params.pkl \
--outfile=/mnt/whatever.pkl \
--n_seqs=10 \
--len_seqs=32