Skip to content

Instantly share code, notes, and snippets.

@koen-dejonghe

koen-dejonghe/am1.scala

Created October 19, 2018 11:42
Show Gist options
  • Save koen-dejonghe/02a1a1fdf45cd1716a6fb1ba8d50f058 to your computer and use it in GitHub Desktop.
Save koen-dejonghe/02a1a1fdf45cd1716a6fb1ba8d50f058 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
am1.scala
val m: Module = new Module() {
val fc = Linear(100, 10)
override def forward(x: Variable): Variable = x ~> fc ~> relu
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment