summaryrefslogtreecommitdiff
path: root/eval.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-01-12 20:23:40 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-01-12 20:23:40 +0800
commit4495e10e5159ee54814e73ae7bfb27db2754ae69 (patch)
treeac2bd997784229d58b72188dfa32d1c1edd0aa57 /eval.py
parent0c027822a48440be59f9b4889831af7879ecf277 (diff)
Add Google TPU support
Diffstat (limited to 'eval.py')
-rw-r--r--eval.py2
1 files changed, 0 insertions, 2 deletions
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 = {