猫狗识别——PyTorch

import os
import time
import numpy as np

import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.utils.data import DataLoader
from torch.utils import data
from torchvision import transforms as T
from PIL import Image

###################################
# SETTINGS
###################################

class Config(object):
    
    batch_size = 32
    max_epoch = 30
    num_workers = 2
    lr = 0.001
    lr_decay = 0.95
    weight_decay = 0.0001
    
    train_data_root = ‘/home/dong/Documents/DATASET/train‘
    test_data_root = ‘/home/dong/Documents/DATASET/test‘
    
    load_dict_path = None

opt = Config()