Skip to content
Snippets Groups Projects
Commit fdf54422 authored by Franck Galpin's avatar Franck Galpin
Browse files

Merge branch 'correct_model_last_prelu' into 'VTM-11.0_nnvc'

remove final prelu

See merge request jvet-ahg-nnvc/VVCSoftware_VTM!101
parents 4d1f6863 0e3cfb9e
No related branches found
No related tags found
No related merge requests found
......@@ -208,7 +208,7 @@ class SADLNet(nn.Sequential):
Conv(D6, C, kernel_size=3, stride=2),
*[ResidualBlock(C, C1, C21, C22, C31) for _ in range(N)],
Conv(C, C, kernel_size=3),
Conv(C, output_channels, kernel_size=3),
Conv(C, output_channels, kernel_size=3, post_activation=None),
)
def get_example_inputs(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment