From 4495e10e5159ee54814e73ae7bfb27db2754ae69 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Tue, 12 Jan 2021 20:23:40 +0800 Subject: Add Google TPU support --- eval.py | 2 -- 1 file changed, 2 deletions(-) (limited to 'eval.py') diff --git a/eval.py b/eval.py index 7b68220..f7fb099 100644 --- a/eval.py +++ b/eval.py @@ -3,9 +3,7 @@ import numpy as np from config import config from models import Model from utils.dataset import ClipConditions -from utils.misc import set_visible_cuda -set_visible_cuda(config['system']) model = Model(config['system'], config['model'], config['hyperparameter']) dataset_selectors = { -- cgit v1.2.3