_
This commit is contained in:
1 parent
47f9bad42b
commit
268b402513
2 files changed
+3
-3
No files matched your search
@@ -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