Pytorch Tricks
- Also look at fastai
Get Params of a Layer
m = learn.model
l = m.get_submodule('0.model.stem.1')
list(l.parameters())Interact
from ipywidgets import interact
@interact(m=1.5, b=1.5)
def plot_relu(m,b):
plot_function(partial(relu, m, b), ylim = (-1, 4))Set Dataset Directory
import os
os.environ["TORCH_HOME"] = "/media/hdd/Datasets/"
os.environ["FASTAI_HOME"] = "/media/hdd/Datasets/"