fix run.sh

This commit is contained in:
bhko 2023-02-14 09:53:50 +00:00
parent ef52abd12d
commit 01b1b22680
3 changed files with 3 additions and 9 deletions

View File

@ -1,4 +1,5 @@
*.pth
*.txt
*.o
*.out
main

View File

@ -25,4 +25,4 @@ clean:
rm -rf $(TARGET) $(OBJECTS)
run: main
sbatch run.sh model/weights.bin img/bin/img1.bin output.bin 1
sbatch run.sh model/weights.bin img/bin/img1.bin output1.bin 1

View File

@ -1,10 +1,3 @@
#!/bin/bash
#SBATCH --job-name=colorizer # Submit a job named "colorizer"
#SBATCH --nodes=1 # Using 1 node
#SBATCH --gpus-per-node=0 # Using 0 GPUs per node
#SBATCH --time=0-00:05:00 # 5 minute timelimit
#SBATCH --mem=16000MB # Using 16GB memory per node
#SBATCH --exclusive # Take node exclusively
./main "$@"
srun -N 1 --partition apws --exclusive --job-name=unet ./main $@