#!/usr/bin/env python3
import argparse
import math

import megengine.functional as F
import megengine.module as M
import numpy as np
from megengine import jit, tensor
class ConvNet(M.Module):
     def __init__(self):

         super().__init__()
         self.conv1 = 
M.Conv2d(in_channels=3, out_channels=1, kernel_size=3, bias=False)