上传者: 42122878
|
上传时间: 2021-12-29 10:58:23
|
文件大小: 1.18GB
|
文件类型: -
神经过程家族
主要入口: 。
什么 ?
该存储库包含:
的文本
Pytorch代码(训练/绘图)以及预训练模型,以研究图像和合成一维数据集上的以下模型:
来自CNP
来自LNP
AttnCNP和AttnLNP
ConvCNP
ConvLNP
(“官方”)代码复制所有图像实验和带。 对于一维实验,请参见和 。
有关如何使用npf库的教程,请参阅NPF网站的“可 再现性”部分。
安装
点子
# clone repo
pip install -r requirements.txt
请注意,skorch的版本必须为0.8,以确保可以正确上传预训练的模型。
码头工人
安装nvidia-docker
使用Dockerfile构建映像或使用Dockerfile docker pull yanndubs/npf:gpu
创建并运行一个容器,例如: docker run --