Add momentum to conv layer with an option true or false