From 7212499f839db946c9296b84d6491fe5d6097dea Mon Sep 17 00:00:00 2001 From: junyanz Date: Tue, 13 Jun 2017 00:10:04 -0700 Subject: predict single image --- train.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) (limited to 'train.py') diff --git a/train.py b/train.py index 12c9dbc..0834d37 100644 --- a/train.py +++ b/train.py @@ -1,11 +1,10 @@ import time from options.train_options import TrainOptions -opt = TrainOptions().parse() # set CUDA_VISIBLE_DEVICES before import torch - from data.data_loader import CreateDataLoader from models.models import create_model from util.visualizer import Visualizer +opt = TrainOptions().parse() data_loader = CreateDataLoader(opt) dataset = data_loader.load_data() dataset_size = len(data_loader) -- cgit v1.2.3-70-g09d2