ENSEEIHT/projet-long
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Remove assert for multiclass

Browse source 
Former-commit-id: 16809ec7ba14f67bfa5e5a095d3b512ef669a964
This commit is contained in:
milesial 2019-12-13 15:58:45 +01:00 committed by GitHub
parent 67c4dd34a0
commit 4e1f0398a1
1 changed files with 0 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
train.py
View file

@ -70,11 +70,6 @@ def train_net(net,
f'but loaded images have {imgs.shape[1]} channels. Please check that ' \
'the images are loaded correctly.'
assert true_masks.shape[1] == net.n_classes, \
f'Network has been defined with {net.n_classes} output classes, ' \
f'but loaded masks have {true_masks.shape[1]} channels. Please check that ' \
'the masks are loaded correctly.'
imgs = imgs.to(device=device, dtype=torch.float32)
true_masks = true_masks.to(device=device, dtype=torch.float32)