feat: added benchmark print for retrain
Some checks failed
continuous-integration/drone/pr Build is failing
Some checks failed
continuous-integration/drone/pr Build is failing
This commit is contained in:
@ -46,6 +46,10 @@ void network_retrain(char *network_path, char *training_data_dir, double batch_p
|
||||
|
||||
train_network(&network, datas, data_len, learning_rate, batch_size, 0, 0, iteration);
|
||||
|
||||
printf("Final network cost: %f\n", get_network_total_cost(&network, datas, data_len));
|
||||
size_t suc = get_network_success_rate(&network, datas, data_len, network.nb_output / 26);
|
||||
printf("Network success rate: %li/%li: %.2f%%\n", suc, data_len, (((double)suc / (double)data_len) * 100.0));
|
||||
|
||||
save_neural_network(&network, network_path);
|
||||
}
|
||||
|
||||
|
Reference in New Issue
Block a user