mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

add rdnet

Browse Source
This commit is contained in:
dong-hyun 2024-08-01 14:51:15 +09:00
parent 225f4f92b3
commit 025259024d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/models/rdnet.py
View File

@ -28,7 +28,7 @@ class Block(nn.Module):
nn.Conv2d(in_chs, in_chs, groups=in_chs, kernel_size=7, stride=1, padding=3), nn.Conv2d(in_chs, in_chs, groups=in_chs, kernel_size=7, stride=1, padding=3),
norm_layer(in_chs), norm_layer(in_chs),
nn.Conv2d(in_chs, inter_chs, kernel_size=1, stride=1, padding=0), nn.Conv2d(in_chs, inter_chs, kernel_size=1, stride=1, padding=0),
act_layer(act_layer), act_layer(),
nn.Conv2d(inter_chs, out_chs, kernel_size=1, stride=1, padding=0), nn.Conv2d(inter_chs, out_chs, kernel_size=1, stride=1, padding=0),
) )