sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-25 10:32:59 +00:00
Code Issues Projects Releases Packages Wiki Activity

allow pretrain to run on gpu

Browse source 
we've slashed the size of the 2nd encoder, so ti should fit naow?
This commit is contained in:
Starbeamrainbowlabs 2022-11-04 17:02:07 +00:00
parent 44ad51f483
commit 0353072d15
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
aimodel/slurm-pretrain.job
View file

@ -5,7 +5,7 @@
#SBATCH --gres=gpu:1
#SBATCH -o %j.%N.%a.out.log
#SBATCH -e %j.%N.%a.err.log
#SBATCH -p gpu05
#SBATCH -p gpu05,gpu
#SBATCH --time=5-00:00:00
#SBATCH --mem=61440
# 61440 = 60GiB memory required