Pytorch: move data into CUDA

move data into CUDA:

cudafy = lambda x : x if cuda is None else x.cuda(cuda)
cudafy(model)

convert data from tensor (in GPU) to numpy (in cpu):

torch2np = lambda x : x.cpu().detach().numpy()
©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容

  • rljs
    rljs by sennchi Timeline of History Part One The Cognitiv...
    sennchi阅读 12,136评论 0赞 10
  • 【转载】Getting started with TensorFlow on iOS
    原文链接：machinethink.net/blog/tensorflow-on-ios/ Before you ...
    dopami阅读 4,629评论 0赞 1
  • 无标题文章
    Effective TensorFlow Table of Contents TensorFlow Basics ...
    法布雷加嘻阅读 3,652评论 0赞 1
  • 天天六一
    看图札记 这两图应该是在昨天发布，但今天也合适：心里有阳光，每天是六一。 从图中看出，人与人之间，需要相互帮衬，但...
    仍需历验阅读 1,241评论 0赞 0
  • 2018-07-27在职读研路
    昨天下午四点，看到人大人经济学专业群里同学们在议论，全国统考成绩出来了。我既有点紧张又期待。 2018年5月，我参...
    宇熙妈妈阅读 1,761评论 0赞 1