def test_train_gdl(self):
dataset = SegmentationDummyDataset(batch_size=32)
pspnet = PSPNet(epochs=1,
dataset=dataset,
frontend_name='resnet101',
generarized_dice_loss={'alpha': 0.01})
history = pspnet.train()
ok_('loss' in history)
def test_train(self):
dataset = SegmentationDummyDataset(batch_size=32)
pspnet = PSPNet(epochs=1, dataset=dataset, frontend_name='resnet101')
history = pspnet.train()
ok_('loss' in history)
