Exemplo n.º 1
0
from segment import *
from MAEC import MAEC
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

data = get_data()
MAEC = MAEC()
MAEC.train(1000, 32, data)