remove explicit device specification

This commit is contained in:
YuDeng 2021-01-27 20:02:06 +08:00 коммит произвёл GitHub
Родитель 1a58957ec1
Коммит c6fd2eb1c5
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 4 добавлений и 3 удалений

Просмотреть файл

@ -58,9 +58,10 @@ def demo():
n = 0
# build reconstruction model
with tf.Graph().as_default() as graph,tf.device('/cpu:0'):
opt = Option(is_train=False)
with tf.Graph().as_default() as graph:
with tf.device('/cpu:0'):
opt = Option(is_train=False)
opt.batch_size = 1
opt.pretrain_weights = args.pretrain_weights
FaceReconstructor = Face3D()