Method
__init__
(self, dim_in, dim_out, act_layer=nn.ReLU, dropout=True, p_dropout_fc=0.25)
Source from the content-addressed store, hash-verified
| 95 | |
| 96 | class FC_block(nn.Module): |
| 97 | def __init__(self, dim_in, dim_out, act_layer=nn.ReLU, dropout=True, p_dropout_fc=0.25): |
| 98 | super(FC_block, self).__init__() |
| 99 | |
| 100 | self.fc = nn.Linear(dim_in, dim_out) |
| 101 | self.act = act_layer() |
| 102 | self.drop = nn.Dropout(p_dropout_fc) if dropout else nn.Identity() |
| 103 | |
| 104 | def forward(self, x): |
| 105 | x = self.fc(x) |
Callers
nothing calls this directly
Tested by
no test coverage detected