Update base.py#333

jzakirov wants to merge 2 commits into

masterphotosynthesis-team/piq:masterfrom

compute_grads_fixphotosynthesis-team/piq:compute_grads_fixCopy head branch name to clipboard