Skip to content

Latest commit

 

History

History

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 
 
 
 
 

README.md

AlexNet

PyTorch を使って AlexNet を実装・学習するための作業ディレクトリです。

Setup

uv sync

Train

uv run python train.py

学習ログを保存する場合:

uv run python train.py | tee training.log

Import check

uv run python -c "import numpy as np; import torch; import torch.nn as nn; from torchvision import datasets, transforms; from torch.utils.data.sampler import SubsetRandomSampler; print(torch.__version__)"