pytorch-glow:OpenAI的GLOW的Pytorch实现

时间:2024-06-05 19:36:53
【文件属性】:

文件名称:pytorch-glow:OpenAI的GLOW的Pytorch实现

文件大小:9KB

文件格式:ZIP

更新时间:2024-06-05 19:36:53

Python

辉光 OpenAI生成模型Pytorch实现。 此仓库提供了一种用于堆叠可逆转换的模块化方法。 运行代码 python train.py 例如 CUDA_VISIBLE_DEVICES=0 python train.py --depth 10 --coupling affine --batch_size 64 --print_every 100 --permutation conv 待办事项 多GPU支持。 如果对您而言性能是一个问题,我建议您签出pytorch实施。 支持更多数据集 LU分解的可逆卷积。 接触 该存储库不再维护。 如果需要,可以随时提出问题,但是响应速度可能很慢。


【文件预览】:
pytorch-glow-master
----test_flows.py(5KB)
----layers.py(2KB)
----utils.py(2KB)
----README.md(735B)
----invertible_layers.py(14KB)
----train.py(7KB)

网友评论