_
This commit is contained in:
@@ -65,9 +65,9 @@ class Model(ModelBase):
|
||||
def onTrainOneIter(self, generators_samples, generators_list):
|
||||
target_src, target_src_mask = generators_samples[0]
|
||||
|
||||
loss,acc = self.fan_seg.train_on_batch( [target_src], [target_src_mask] )
|
||||
loss = self.fan_seg.train_on_batch( [target_src], [target_src_mask] )
|
||||
|
||||
return ( ('loss', loss), ('acc',acc))
|
||||
return ( ('loss', loss), )
|
||||
|
||||
#override
|
||||
def onGetPreview(self, sample):
|
||||
|
||||
Reference in New Issue
Block a user