time | Calls | line |
---|
| | 67 | function model = update_weights(model, alpha, x)
|
| | 68 |
|
< 0.001 | 5 | 69 | if alpha == 1
|
< 0.001 | 5 | 70 | old_alpha = 0;
|
| | 71 | else
|
| | 72 | old_alpha = alpha / 0.5;
|
< 0.001 | 5 | 73 | end
|
| | 74 |
|
< 0.001 | 5 | 75 | var_ptr = model.var_ptr;
|
0.001 | 5 | 76 | channel_and_neurons = [model.ch_input; model.full_neurons];
|
0.001 | 5 | 77 | for m = 1 : model.L
|
0.002 | 20 | 78 | range = var_ptr(m):var_ptr(m+1)-1;
|
0.004 | 20 | 79 | X = reshape(x(range), channel_and_neurons(m+1), []);
|
0.005 | 20 | 80 | model.weight{m} = model.weight{m} + (alpha - old_alpha) * X(:, 1:end-1);
|
0.002 | 20 | 81 | model.bias{m} = model.bias{m} + (alpha - old_alpha) * X(:, end);
|
< 0.001 | 20 | 82 | end
|
Other subfunctions in this file are not included in this listing.