Exemplo n.º 1
0
 def loss_func(*params):
     f = model.forward_batch(batch, 'train')
     return model.loss(f, label)
Exemplo n.º 2
0
 def loss_func(*params):
     f = model.forward(img, 'train')
     return model.loss(f, label)
Exemplo n.º 3
0
 def loss_func(*params):
     f = model.forward_batch(batch, 'train')
     return model.loss(f, label)