Method
__init__
(self, net_enc, net_dec, crit, deep_sup_scale=None)
Source from the content-addressed store, hash-verified
| 20 | |
| 21 | class SegmentationModule(SegmentationModuleBase): |
| 22 | def __init__(self, net_enc, net_dec, crit, deep_sup_scale=None): |
| 23 | super(SegmentationModule, self).__init__() |
| 24 | self.encoder = net_enc |
| 25 | self.decoder = net_dec |
| 26 | self.crit = crit |
| 27 | self.deep_sup_scale = deep_sup_scale |
| 28 | |
| 29 | def forward(self, feed_dict, *, segSize=None): |
| 30 | # training |
Callers
nothing calls this directly
Tested by
no test coverage detected