update dataset_loader

2018-03-12 18:39:28 +00:00 · 2018-03-12 18:39:28 +00:00 · b0513b6b6e
parent 4ab5445ddb
commit b0513b6b6e
1 changed files with 1 additions and 0 deletions
--- a/dataset_loader.py
+++ b/dataset_loader.py
@ -3,6 +3,7 @@ import os
 from PIL import Image
 import numpy as np

+import torch
 from torch.utils.data import Dataset

 def read_image(img_path):