launch_model.sh
No OneTemporary
Actions

Subscribers

None

File Metadata

	#!/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)