Flatten weights and fill! were allocating memory, devectorize them
More timings and fix bugs'
Removing debug print statement from log_loss