Research/CDAL

[CDAL Research] CNN Model for Image Classification

1. Module Import & GPU Usage Basic Module import import torch import torch.nn as nn import torch.nn.functional as F import torch.optim as optim from torchvision import datasets, transforms from torchvision.transforms import autoaugment from torch.utils.data import DataLoader, random_split import numpy as np import os from PIL import Image import matplotlib.pyplot as plt M1 Macbook에서 GPU 사용: mps ..