jittor和pytorch生成网络对比之pixelda

jittor代码

from __future__ import print_function

import errno
import os
import pickle
from jittor.dataset.dataset import Dataset
from PIL import Image


class MNISTM(Dataset):
    """`MNIST-M Dataset."""

    def __init__<