【文件属性】:
文件名称:flow-based-CoopNets
文件大小:19KB
文件格式:ZIP
更新时间:2021-03-31 02:10:13
Python
基于流的CoopNets
用法
环境设定
确保已安装 。
使用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 train.py进行默认配置,或者运行python train.py --gpu_ids=0,1在2个GPU上运行,而不是在1个GPU上运行。 这也将使批次大小增加一倍。
在每个纪元末尾,来自模型的样本将保存到samples/e