Skip to content

Commit

Permalink
Fix FCResidualBlock by removing relu (#368)
Browse files Browse the repository at this point in the history
<img width="785" alt="Screenshot 2024-02-29 at 1 08 04 PM"
src="https://github.com/pyg-team/pytorch-frame/assets/16249234/c3e7940b-3b62-4143-91f7-da8a44d03bb1">
  • Loading branch information
weihua916 committed Feb 29, 2024
1 parent e90f32a commit b8d17b6
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 1 deletion.
1 change: 1 addition & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
### Changed

- Changed a workflow of Encoder's `na_forward` method resulting in performance boost ([#364](https://github.com/pyg-team/pytorch-frame/pull/364))
- Removed ReLU applied in `FCResidualBlock` ([#368](https://github.com/pyg-team/pytorch-frame/pull/368))

### Deprecated

Expand Down
1 change: 0 additions & 1 deletion torch_frame/nn/models/resnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,6 @@ def forward(self, x: Tensor) -> Tensor:
x = self.shortcut(x)

out = out + x
out = self.relu(out)

return out

Expand Down

0 comments on commit b8d17b6

Please sign in to comment.