Make each cell more independent

[Michael-Equi]

Either init all variables for a cell in the cell OR have one cell that initiates everything

[Michael-Equi]

For example, in order to use grad cam on the ipynb I need to jump around to init everything in different places before it just works