Skip to content

Commit

Permalink
WIP: add workflow to run prediction on lidar-prod optimization dataset
Browse files Browse the repository at this point in the history
  • Loading branch information
leavauchier committed Apr 18, 2024
1 parent fef936f commit 76f5094
Showing 1 changed file with 76 additions and 0 deletions.
76 changes: 76 additions & 0 deletions .github/workflows/predict-for-lidar-prod-optimization.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,76 @@
# Workflow name
name: "Prediction on lidar-prod optimization dataset"

on:
# Run workflow on user request
workflow_dispatch:
inputs:
sampling_name:
description: >
Sampling name (nom du dataset sur lequel le modèle a été entraîné).
Eg. YYYYMMDD_MonBeauDataset
required: true
user:
description: >
Username, utilisé pour générer un chemin standard pour les sorties dans le
dossier IA du store (projet-LHD/IA/MYRIA3D-SHARED-WORKSPACE/$USER/$SAMPLING_NAME/)
required: true
checkpoint_name:
description: >
Nom du checkpoint (sans l'extension) ) utiliser pour les prédictions
Exemple : 20230930_60k_basic_targetted_epoch37_Myria3DV3.4.0
required: true
predict_config_name:
description: >
Nom du fichier de config de myriad utilisé pour la prédiction
(doit exister dans projet-LHD/IA/MYRIA3D-SHARED-WORKSPACE/$USER/$SAMPLING_NAME/)
Exemple: 20230930_60k_basic_targetted_epoch37_Myria3DV3.4.0_predict_config_V3.5.0.yaml
required: true

env:
WORKDIR: /var/data/LIDAR-PROD-OPTIMIZATION/
DATA: ${{ WORKDIR }}/20221018_lidar-prod-optimization-on-151-proto/Comparison/
LOGS_DIR: /var/data/MYRIA3D-SHARED-WORKSPACE/$USER/$SAMPLING_NAME/"

jobs:
print-inputs:
runs-on: self-hosted
steps:
- run: |
echo "Run prediction on lidar-prod optimzation datasets (val and test)"
echo "Sampling name: ${{ github.event.inputs.sampling_name }}"
echo "User name: ${{ github.event.inputs.user }}"
echo "checkpoint name ${{ github.event.inputs.checkpoint_name }}"
echo "predict_config_name name ${{ github.event.inputs.predict_config_name }}"
echo "workdir: ${{env.WORKDIR}}"
echo "workdir: ${{env.DATA}}"
echo "workdir: ${{env.LOG_DIR}}"
# predict-validation-dataset:
# runs-on: self-hosted

# steps:
# - name: Checkout branch
# uses: actions/checkout@v4

# # See https://github.com/marketplace/actions/setup-micromamba
# - name: setup-micromamba
# uses: mamba-org/[email protected]
# with:
# environment-file: environment.yml
# environment-name: myria3d # activate the environment
# cache-environment: true
# cache-downloads: true
# generate-run-shell: true

# - name: Run prediction on validation dataset
# run: >
# python run.py
# --config-path ${LOGS_DIR}
# --config-name ${LOGS_DIR}${PREDICT_CONFIG_NAME}
# task.task_name=predict
# predict.src_las=${DATA}/val/*.laz
# model.ckpt_path=${LOGS_DIR}${CHECKPOINT_NAME}.ckpt
# predict.output_dir=${WORKDIR}/${SAMPLING_NAME}/preds-valset/
# predict.interpolator.probas_to_save=[building]
# predict.gpus=[0]

0 comments on commit 76f5094

Please sign in to comment.