Skip to content

有人通过run.py成功训练么? #10

@wuzhh8

Description

@wuzhh8

moderntcn.py 中,218行的代码,缩进是不是应该for循环内?这么改的话 运行时会报index out of range的错误。改过之后 还是会报错出现mat1和mat2不匹配的问题。
214 if self.num_stage > 1:
for i in range(self.num_stage-1):
downsample_layer = nn.Sequential(
nn.BatchNorm1d(dims[i]),
nn.Conv1d(dims[i], dims[i + 1], kernel_size=downsample_ratio, stride=downsample_ratio),
)
218 self.downsample_layers.append(downsample_layer)

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