文件名称:flow-based-CoopNets
文件大小:19KB
文件格式:ZIP
更新时间:2024-04-19 19:56:53
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