fix run.sh
This commit is contained in:
parent
ef52abd12d
commit
01b1b22680
|
@ -1,4 +1,5 @@
|
|||
*.pth
|
||||
*.txt
|
||||
*.o
|
||||
*.out
|
||||
main
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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 $@
|
||||
|
|
Loading…
Reference in New Issue