Code

load model to cpu when no cuda#7

Code

Closed

szerintedmi wants to merge 2 commits intoxuebinqin:masterxuebinqin/U-2-Net:masterfrom szerintedmi:no_cuda_fixszerintedmi/U-2-Net:no_cuda_fixCopy head branch name to clipboard

+4-1

Commits

Commits on May 11, 2020

• load model to cpu when no cuda

  szerintedmi

  committed
  1b247c5

  Copy full SHA for 1b247c5

• map location to work on torch < 4.0.1

  szerintedmi

  committed
  4031385

  Copy full SHA for 4031385