Small fixes for notebook to run without issue.#3

ebezzam wants to merge 1 commit intoWaller-Lab:masterWaller-Lab/LenslessLearning:masterfrom

ebezzam:small_fixesebezzam/LenslessLearning:small_fixesCopy head branch name to clipboard