| | |
| | | if(count == 0){ |
| | | net->batch = option_find_int(options, "batch",1); |
| | | input = option_find_int(options, "input",1); |
| | | float learning_rate = option_find_float(options, "learning_rate", .001); |
| | | float momentum = option_find_float(options, "momentum", .9); |
| | | float decay = option_find_float(options, "decay", .0001); |
| | | net->learning_rate = learning_rate; |
| | | net->momentum = momentum; |
| | | net->decay = decay; |
| | | }else{ |
| | | input = get_network_output_size_layer(*net, count-1); |
| | | } |
| | |
| | | strip(line); |
| | | switch(line[0]){ |
| | | case '[': |
| | | printf("%s\n", line); |
| | | current = malloc(sizeof(section)); |
| | | list_insert(sections, current); |
| | | current->options = make_list(); |