add class for overwriting grad
add class for overwriting grad

diogo149/treeano,diogo149/treeano,nsauder/treeano,nsauder/treeano,diogo149/treeano,jagill/treeano,jagill/treeano,nsauder/treeano,jagill/treeano