Skip to content

Instantly share code, notes, and snippets.

View PrimeF's full-sized avatar

Fabrizio Primerano PrimeF

  • Heidelberg (Germany)
View GitHub Profile
---
optimizer:
class: "torch.optim.SGD"
params:
lr: 0.1
momentum: 0.9
nesterov: True
loss: "torch.nn.MSELoss"
inner_activation: "torch.nn.ReLU"
outer_activation: "torch.nn.Softsign"