#!/usr/bin/env python3
from dataset import SIDDValData
from model import UNetD
import megengine.data as data
from utils import batch_PSNR
from tqdm import tqdm
import argparse
import pickle
import megengine


def test(args):
    valid_dataset = SIDDValData(args.data)
    valid_sampler = 
data.SequentialSampler(
        valid_dataset, batch_size=1, drop_la