LBANN(Livermore Big Artificial Neural Network) 툴킷은 여러 수준의 병렬 처리를 구성하도록 최적화된 오픈 소스 HPC 중심 딥 러닝 교육 프레임워크입니다.
LBANN은 네트워크 훈련의 강력한 확장을 최적화하기 위해 도메인 분해를 통해 모델 병렬 가속을 제공합니다. 또한 대량의 데이터로 대규모 신경망을 훈련하기 위해 데이터 병렬성과 앙상블 훈련 방법을 모두 사용하여 모델 병렬성을 구성할 수 있습니다. LBANN은 긴밀하게 결합된 가속기, 낮은 대기 시간의 고대역폭 네트워킹 및 고대역폭 병렬 파일 시스템을 활용할 수 있습니다.
LBANN은 기존의 지도 학습 외에 비지도, 자기 지도, 적대적(GAN) 훈련 방법 등 최첨단 훈련 알고리즘을 지원합니다. 또한 BPTT(역전파) 훈련, 전이 학습, 다중 모델 및 앙상블 훈련 방법을 통해 순환 신경망을 지원합니다.
LBANN 사용자가 LBANN을 설치하는 데 선호되는 방법은 Spack을 사용하는 것입니다. 일부 시스템 구성 후에는 다음과 같이 간단해집니다.
spack install lbann
LBANN 구축 및 설치에 대한 자세한 지침은 기본 LBANN 설명서에서 확인할 수 있습니다.
LBANN을 실행하기 위한 기본 템플릿은 다음과 같습니다.
< mpi-launcher > < mpi-options >
lbann < lbann-options >
--model=model.prototext
--optimizer=opt.prototext
--reader=data_reader.prototext
GPGPU 가속기를 사용할 때 사용자는 LBANN이 MPI 순위 당 하나의 GPU를 할당하는 경우에 최적화되어 있다는 점을 알아야 합니다. MPI 실행 프로그램의 매개변수를 선택할 때 이 점을 염두에 두어야 합니다.
LBANN 실행에 대한 자세한 내용은 여기에 설명되어 있습니다.
출판물, 프리젠테이션, 포스터 목록이 여기에 표시됩니다.
문제, 질문, 버그는 Github 문제 추적기에서 제기할 수 있습니다.