EnergyMatching
[NeurIPS 2025] Official repository for "Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative Modeling"
파일 탐색기
최종 버전 다운로드 (.zip)- __init__.py
- config_multigpu.py
- fid_cifar_heun_1gpu.py
- network_transformer_vit.py
- sample_cifar_heun_1gpu.py
- train_cifar_multigpu.py
- config_multigpu_imagenet32.py
- dataset_imagenet32.py
- fid_imagenet_heun_multigpu.py
- network_transformer_vit.py
- train_imagenet_multigpu.py
- aav_ground_truth.csv
- aav_hard.csv
- aav_medium.csv
- config_aav.yaml
- oracle_aav.ckpt
- predictor_aav_hard.ckpt
- predictor_aav_medium.ckpt
- predictor_aav_hard.ckpt
- predictor_aav_medium.ckpt
- vae_aav_hard.pt
- vae_aav_medium.pt
- __init__.py
- config.py
- model_proteins.py
- oracle.py
- sampling.py
- train_proteins.py
- utils_proteins.py
- utils_train_proteins.py
- __init__.py
- tutorial_2D.ipynb
- utils_2D.py
- __init__.py
- cifar10_FID_3_3.gif
- EM_2D.png
- EnergyMatchingTalk.pdf
- LICENSE
- README.md
- requirements.txt
- utils_cifar_imagenet.py
# 설치 가이드
1. 코드 내려받기
git clone https://github.com/m1balcerak/EnergyMatching
깃허브에서 프로젝트 코드 전체를 내 컴퓨터로 내려받습니다.
cd EnergyMatching
방금 내려받은 프로젝트 폴더 안으로 이동합니다.
2. 공식 설치 스크립트
쉬움 추천사전 준비물
- Python 3 pip 명령어를 쓰려면 Python이 필요합니다.
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
PyPI에 배포된 패키지를 바로 설치합니다. 소스 클론이 필요 없습니다.
설치 후 새 터미널을 열고, 프로그램의 버전 확인 명령(예: --version)으로 정상 설치됐는지 확인하세요.
이 레포의 README에 적힌 실제 명령어를 그대로 가져왔습니다.
3. Python
쉬움사전 준비물
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
PyPI에 배포된 패키지를 바로 설치합니다. 소스 클론이 필요 없습니다.
pip install -r requirements.txt
requirements.txt 등에 명시된 파이썬 라이브러리를 설치합니다.
에러 메시지 없이 실행되고 터미널에 안내 문구가 출력되면 정상입니다.
이 레포의 README에 적힌 실제 명령어를 그대로 가져왔습니다.
// repository documentation
Was this content helpful?
(0 ratings)