Commit b3b48cd3 authored by Sensing's avatar Sensing

Update train_resize_index.py

parent 4eb1a545
......@@ -6,7 +6,8 @@ import normalization
import cv2
import math
import os
os.environ['CUDA_VISIBLE_DEVICES']='2,3'
os.environ['CUDA_VISIBLE_DEVICES']='2'
class autoencoder():
def __init__(
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