fix the problem with reading the weights in parallel, add notes from the review