在PyTorch中发光
在PyTorch中实现Glow。 根据论文:
Diederik P. Kingma,Prafulla Dhariwal arXiv:1807.03039
设计训练脚本和超参数以匹配本文表4中描述的CIFAR-10实验。
用法
环境设定
确保已安装 。
使用git clone https://github.com/chrischute/glow.git glow克隆克隆回购。
进入克隆的仓库: cd glow 。
创建环境: conda env create -f environment.yml 。
激活环境: source activate glow 。
火车
确保已如上所述创建并激活了conda环境。
运行python train.py -h以查看选项。
运行python train.py [FLAGS]进行训练。 例如,运行python tra
2021-11-10 15:51:43
11.18MB
Python
1