jittor代码

import jittor as jtfrom jittor import initfrom jittor import nnfrom jittor.dataset.mnist import MNISTimport jittor.transform as transformimport argparseimport osimport numpy as npimport mathimport cv2import time

jt