launch_model.sh
e8b57b448f0crc_developments

launch_model.sh
View Options

			#!/bin/bash

			## ====== Settings

			#SBATCH --account=cadmos

			## specific for GPU nodes (free account)
			#SBATCH --partition=gpu --qos=gpu --gres=gpu:4

			#SBATCH --nodes=4
			#SBATCH --mem=25G
			#SBATCH --ntasks=4
			#SBATCH --cpus-per-task=4
			#SBATCH --time=12:00:00
			##SBATCH -o UNet_training.out # Standard output
			##SBATCH -e UNet_training.err # Standard error


			#SBATCH --workdir /home/rcastell/solar_deployment/segmentation/CNN/logs

			#SBATCH --mail-user=roberto.castello@epfl.ch

			echo STARTING AT $(date)

			## == Specify here the needed modules and the virtual env

			module purge
			module load gcc cuda cudnn python
			source /home/rcastell/solar_deployment/1.9-gpu/tensorflow-1.9/bin/activate

			## == Specify here the commands to execute

			## Run. Argument to pass (in order): rotations, lights, dice, weightnopv, weightpv, epochs
			python $HOME/solar_deployment/segmentation/CNN/model_main.py 0 0 0 1 3.5 30

			source deactivate

			echo FINISHED AT $(date)