predict single image

1 files changed, 1 insertions, 2 deletions

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)