Kaixhin/PlaNet

Nonlinearity flag

Kaixhin opened this issue · 0 comments

Kaixhin commented

Add flag to choose nonlinearity from torch.nn.functional, defaulting to relu.