Skip to content
Snippets Groups Projects
Commit a34560b0 authored by Pedro Gonnet's avatar Pedro Gonnet
Browse files

fix really, really stupid bug. got the re-scaling of the graph weights backwards.

Former-commit-id: b0776d4e3f2c4a6bbb463bb384b1c63b2c47fb3e
parent 0f494287
No related branches found
No related tags found
No related merge requests found
...@@ -443,7 +443,7 @@ void engine_repartition ( struct engine *e ) { ...@@ -443,7 +443,7 @@ void engine_repartition ( struct engine *e ) {
error( "Failed to allreduce the weight scales (%s)." , buff ); error( "Failed to allreduce the weight scales (%s)." , buff );
} }
if (wscale_buff != wscale) { if (wscale_buff != wscale) {
float scale = wscale / wscale_buff; float scale = wscale_buff / wscale;
for (k = 0; k < 26 * nr_cells; k++) weights_e[k] *= scale; for (k = 0; k < 26 * nr_cells; k++) weights_e[k] *= scale;
for (k = 0; k < nr_cells; k++) weights_v[k] *= scale; for (k = 0; k < nr_cells; k++) weights_v[k] *= scale;
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment