Parameter use_gpu: true + #SBATCH --partition=p.hpcl93 however still cpu is used for transfer learning.