torch量化感知训练示例项目

Posted 2021-06-30 踟蹰横渡口,彳亍上滩舟。

1. 依赖相关库文件

import numpy as np
import torch
import torch.nn as nn
import torchvision
from torch.utils.data import DataLoader
from torchvision import datasets
import torchvision.transforms as transforms
import os
import time
import sys