Add Google TPU support

1 files changed, 0 insertions, 2 deletions

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