MLproject

# Author: Lukas Heumos
# Email: lukas.heumos@posteo.net
# Version: 1.0.0
# License: MIT


name: sc-autoencoder

# conda_env: environment.yml
docker_env:
  image: ghcr.io/mlf-core/sc_autoencoder:1.0.0
  volumes: ["${PWD}/data:/data"]

entry_points:
  main:
    parameters:
      cuda: {type: string, default: 'True'}

      max_epochs: {type: int, default: 1000}
      general-seed: {type: int, default: 0}
      tensorflow-seed: {type: int, default: 0}
      batch-size: {type: int, default: 256}
      lr: {type: float, default: 0.001}
      mixed-precision: {type: string, default: 'False'}
      deterministic: {type: string, default: '1'}

    command: |
          python sc_autoencoder/sc_autoencoder.py \
            --cuda {cuda} \
            --max_epochs {max_epochs} \
            --general-seed {general-seed} \
            --tensorflow-seed {tensorflow-seed} \
            --batch-size {batch-size} \
            --lr {lr} \
            --mixed-precision {mixed-precision} \
            --deterministic {deterministic}