Skip to content

corrections to help people reproduce #34

@coco515alt

Description

@coco515alt
  1. requirements不全,缺少lpips库,并且tensorboard最新版与现有环境似乎有版本冲突。
  2. 参数类型设置有误,如果是单卡的情况下需要把is_module设置为False,但是这个参数会把False解析成字符串造成训练流程中判断错误,进入语义优化阶段的时候会中断训练流程。单卡情况下直接将.sh文件这一行参数删去(默认值为False),想要修正的话在training_utils文件中补充函数str2bool,并将这个参数的type设置为str2bool。
  3. 参数的具体设置在training_utils中,可以将参数单独写成一个配置文件直接读取,方便修改调试。需要注意的是默认最大训练步骤是100k,论文中语义阶段的优化步骤数量为8.5k,所以需要把最大训练步骤改成12.5k。

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions