Skip to content

Instantly share code, notes, and snippets.

@cgarciae
Last active August 25, 2021 20:01
Show Gist options
  • Save cgarciae/bbfaeeb97a63916fc9d3bcad022310ad to your computer and use it in GitHub Desktop.
Save cgarciae/bbfaeeb97a63916fc9d3bcad022310ad to your computer and use it in GitHub Desktop.
Module print
MyModule:
b: list
- Initializer
- Parameter((5, 13), float32)
a: dict
mlps: list
- MLP:
linear1: Linear
w: Parameter((2, 3), float32)
b: Parameter((3,), float32)
n: Nothing
linear2: Linear
w: Parameter((3, 5), float32)
b: Parameter((5,), float32)
n: Nothing
- MLP:
linear1: Linear
w: Parameter((2, 3), float32)
b: Parameter((3,), float32)
n: Nothing
linear2: Linear
w: Parameter((3, 5), float32)
b: Parameter((5,), float32)
n: Nothing
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment