senet.pytorch

(★ 2,344)

PyTorch implementation of SENet

오너의 다른 레포

ewc.pytorch ★ 259

An implementation of EWC with PyTorch

Jupyter Notebook
sam.pytorch ★ 136

A PyTorch implementation of Sharpness-Aware Minimization for Efficiently Improving Generalization

Python
dda ★ 125

Differentiable Data Augmentation Library

Jupyter Notebook
homura ★ 104

homura is a library for fast prototyping DL research

Python
shampoo.pytorch ★ 79

An implementation of shampoo

Python
cca.pytorch ★ 70

CCAs for looking into DNNs

Python

파일 탐색기

ZIP 다운로드
  • .gitignore
  • cifar.py
  • hubconf.py
  • imagenet.py
  • LICENSE
  • README.md

# 설치 가이드

python
사전 준비물

설치 및 실행 절차

pip install git+https://github.com/moskomule/homura@v2020.07

프로젝트 실행에 필요한 homura 라이브러리를 설치합니다.

python cifar.py

Cifar10 데이터셋으로 SE-ResNet20 모델을 실행합니다.

핵심 명령어

pip install git+https://github.com/moskomule/homura@v2020.07

Homura 라이브러리를 깃허브에서 직접 설치합니다.

python cifar.py

CIFAR-10 데이터셋을 이용해 SE-ResNet20을 실행합니다.

python imagenet.py

ImageNet 데이터셋을 이용해 SE-ResNet50 학습/실행을 수행합니다.

python -m torch.distributed.launch --nproc_per_node=2 imagenet.py

다중 GPU 환경에서 분산 학습을 실행합니다.

CIFAR-10과 ImageNet 데이터셋은 ~/.torch/data 경로에 위치해야 하며, ImageNet은 IMAGENET_ROOT 환경 변수로 지정할 수 있습니다.

# CDN으로 사용하기

jsDelivr

jsDelivr는 공개 GitHub 리포지토리를 별도 설정 없이 CDN으로 즉시 서빙합니다. 버전과 파일을 고르면 웹페이지에 바로 붙일 수 있는 링크와 예시 코드가 만들어집니다.

링크
예시

명령어 용어집

이 문서에서 사용된 명령어를 모아봤습니다. 낯선 명령어가 있다면 펼쳐서 확인해보세요.

🔍

pip install

설명 보기 ▼

Install Python packages.

pip install {{package1 package2 ...}}

Install one or more packages:

pip install {{package1 package2 ...}} {{[-U|--upgrade]}}

Upgrade all specified packages to the latest version, installing any that are not already present:

pip install {{package}}=={{version}}

Install a specific version of a package:

🔍

python

설명 보기 ▼

Python 언어 인터프리터.
더 많은 정보: <https://docs.python.org/using/cmdline.html>.

python

REPL(대화형 셸) 시작:

python {{path/to/file.py}}

특정 Python 파일 실행:

python -i {{path/to/file.py}}

특정 Python 파일 실행 후 REPL 시작:

// repository documentation