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
......@@ -443,7 +443,7 @@ void engine_repartition ( struct engine *e ) {
error( "Failed to allreduce the weight scales (%s)." , buff );
}
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 < nr_cells; k++) weights_v[k] *= scale;
}
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment