from secure_ml.utils.utils import DataSet, try_gpu
from secure_ml.utils.utils import Conv2d, ConvTranspose2d

from secure_ml.utils.utils import DataSet, try_gpu
from secure_ml.utils.utils import Conv2d, ConvTranspose2d
