what's in loss again
This commit is contained in:
1 parent
0b7a20c651
commit
5ee171689c
1 file changed
+1
-1
+1
-1
@@ -873,7 +873,7 @@ def main():
|
||||
logger.info("HHHHH Forward")
|
||||
loss, _ = model(input_ids, segment_ids, input_mask, start_positions, end_positions)
|
||||
model.zero_grad()
|
||||
logger.info("HHHHH Backward")
|
||||
logger.info("HHHHH Backward, loss: {}".format(loss))
|
||||
loss.backward()
|
||||
logger.info("HHHHH Loading data")
|
||||
optimizer.step()
|
||||
|
||||
Reference in new issue
Block a user