elepp

Results 1 comments of elepp

use `criterion_segment = torch.nn.BCEWithLogitsLoss()` and take away `nn.ReLU(inplace=True)` in models.py at `self.layer5 = nn.Sequential( nn.Conv2d(1024, 1, 1), nn.ReLU(inplace=True) )`