Commit b3b48cd3 authored by Sensing's avatar Sensing

Update train_resize_index.py

parent 4eb1a545
...@@ -6,7 +6,8 @@ import normalization ...@@ -6,7 +6,8 @@ import normalization
import cv2 import cv2
import math import math
import os import os
os.environ['CUDA_VISIBLE_DEVICES']='2,3' os.environ['CUDA_VISIBLE_DEVICES']='2'
class autoencoder(): class autoencoder():
def __init__( def __init__(
self, self,
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment