Model name

Environment

pytorch
batch first
include training function
include save and load
include predict

Customize view

indigo
outlined
Dark theme
Show names
Show shapes