Cpu Not Being Used During Training