From 4495e10e5159ee54814e73ae7bfb27db2754ae69 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Tue, 12 Jan 2021 20:23:40 +0800
Subject: Add Google TPU support

---
 train.py | 2 --
 1 file changed, 2 deletions(-)

(limited to 'train.py')

diff --git a/train.py b/train.py
index d91dcd0..e845d6f 100644
--- a/train.py
+++ b/train.py
@@ -1,9 +1,7 @@
 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'])
 
 # 3 models for different conditions
-- 
cgit v1.2.3