74 lines
2.0 KiB
Bash
Executable File
74 lines
2.0 KiB
Bash
Executable File
#! /bin/bash
|
|
#SBATCH -N 1
|
|
#SBATCH --array=0
|
|
#SBATCH -J magic123
|
|
#SBATCH -o slurm_logs/%x.%3a.%A.out
|
|
#SBATCH -e slurm_logs/%x.%3a.%A.err
|
|
#SBATCH --time=3:00:00
|
|
#SBATCH --gres=gpu:v100:1
|
|
#SBATCH --cpus-per-gpu=6
|
|
#SBATCH --mem=30G
|
|
##SBATCH --gpus=1
|
|
|
|
module load gcc/7.5.0
|
|
|
|
|
|
#source ~/.bashrc
|
|
#source activate magic123
|
|
source venv_magic123/bin/activate
|
|
which python
|
|
|
|
nvidia-smi
|
|
nvcc --version
|
|
|
|
hostname
|
|
NUM_GPU_AVAILABLE=`nvidia-smi --query-gpu=name --format=csv,noheader | wc -l`
|
|
echo "number of gpus:" $NUM_GPU_AVAILABLE
|
|
|
|
RUN_ID=$2
|
|
RUN_ID2=$3
|
|
DATA_DIR=$4
|
|
IMAGE_NAME=rgba.png
|
|
step1=$5
|
|
step2=$6
|
|
FILENAME=$(basename $DATA_DIR)
|
|
dataset=$(basename $(dirname $DATA_DIR))
|
|
echo reconstruct $FILENAME under dataset $dataset from folder $DATA_DIR ...
|
|
|
|
if (( ${step1} )); then
|
|
CUDA_VISIBLE_DEVICES=$1 python main.py -O \
|
|
--text "A high-resolution DSLR image" \
|
|
--sd_version 1.5 \
|
|
--image ${DATA_DIR}/${IMAGE_NAME} \
|
|
--workspace out/magic123-3d/magic123-3d-${RUN_ID}/$dataset/magic123-3d_${FILENAME}_${RUN_ID}_5k \
|
|
--optim adam \
|
|
--iters 5000 \
|
|
--guidance zero123 \
|
|
--lambda_guidance 40 \
|
|
--guidance_scale 5 \
|
|
--latent_iter_ratio 0 \
|
|
--normal_iter_ratio 0.2 \
|
|
--t_range 0.2 0.6 \
|
|
--bg_radius -1 \
|
|
--save_mesh \
|
|
${@:7}
|
|
fi
|
|
|
|
if (( ${step2} )); then
|
|
CUDA_VISIBLE_DEVICES=$1 python main.py -O \
|
|
--text "A high-resolution DSLR image" \
|
|
--sd_version 1.5 \
|
|
--image ${DATA_DIR}/${IMAGE_NAME} \
|
|
--workspace out/magic123-3d/magic123-3d-${RUN_ID}-${RUN_ID2}/$dataset/magic123-3d_${FILENAME}_${RUN_ID}_${RUN_ID2} \
|
|
--dmtet --init_ckpt out/magic123-3d/magic123-3d-${RUN_ID}/$dataset/magic123-3d_${FILENAME}_${RUN_ID}_5k/checkpoints/magic123-3d_${FILENAME}_${RUN_ID}_5k.pth \
|
|
--iters 5000 \
|
|
--optim adam \
|
|
--latent_iter_ratio 0 \
|
|
--guidance zero123 \
|
|
--lambda_guidance 0.01 \
|
|
--guidance_scale 5 \
|
|
--bg_radius -1 \
|
|
--rm_edge \
|
|
--save_mesh
|
|
fi
|