Skip to content

resnet_model maybe wrong in resnet_v1 #4198

@irmowan

Description

@irmowan

When I use building_block_v1 or bottleneck_block_v1, the last several layers are:

block: Conv, Batch Norm, add, ReLU, (https://github.com/tensorflow/models/blob/master/official/resnet/resnet_model.py#L143
)
then, Batch Norm, ReLU (https://github.com/tensorflow/models/blob/master/official/resnet/resnet_model.py#L521)

This seems wrong. The last BN and ReLU appears only in Resnet_v2.

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions