good chance I didn't break anything

This commit is contained in:
Joseph Redmon
2016-09-12 13:55:20 -07:00
parent 8ec889f103
commit 5c067dc447
19 changed files with 558 additions and 298 deletions

View File

@ -318,11 +318,11 @@ void backward_network(network net, network_state state)
float train_network_datum(network net, float *x, float *y)
{
*net.seen += net.batch;
#ifdef GPU
if(gpu_index >= 0) return train_network_datum_gpu(net, x, y);
#endif
network_state state;
*net.seen += net.batch;
state.index = 0;
state.net = net;
state.input = x;