(train_ix, global_step, training_steps, current_lr, loss_value)
| 59 | |
| 60 | |
| 61 | def log_callback_st(train_ix, global_step, training_steps, current_lr, loss_value): |
| 62 | logs_writer.add_scalar('train_loss', loss_value, global_step) |
| 63 | logs_writer.add_scalar('lr', current_lr[0], global_step) |
| 64 | |
| 65 | |
| 66 | # Model definition |
nothing calls this directly
no outgoing calls
no test coverage detected