Skip to content

corrections to help people reproduce #34

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。

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

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