Преглед изворни кода

update test.py model.half()

5.0
Glenn Jocher пре 4 година
родитељ
комит
0afbb8d498
1 измењених фајлова са 2 додато и 2 уклоњено
  1. +2
    -2
      test.py

+ 2
- 2
test.py Прегледај датотеку

# model = nn.DataParallel(model) # model = nn.DataParallel(model)


# Half # Half
half = device.type != 'cpu' and torch.cuda.device_count() == 1 # half precision only supported on single-GPU
half = device.type != 'cpu' # half precision only supported on CUDA
if half: if half:
model.half() # to FP16
model.half()


# Configure # Configure
model.eval() model.eval()

Loading…
Откажи
Сачувај