Skip to content

Instantly share code, notes, and snippets.

@joannapurosto

joannapurosto/model-summary.sh

Created August 29, 2019 09:48
Show Gist options
  • Save joannapurosto/d46a4e731936cfa264c42b7b89399c05 to your computer and use it in GitHub Desktop.
Save joannapurosto/d46a4e731936cfa264c42b7b89399c05 to your computer and use it in GitHub Desktop.
Download ZIP
How to do deep learning for java on the Valohai platform? -blog | Valohai
Raw
model-summary.sh
[<--- snipped --->]
11:17:05 =========================================================================
11:17:05 LayerName (LayerType) nIn,nOut TotalParams ParamsShape
11:17:05 =========================================================================
11:17:05 layer0 (DenseLayer) 784,1000 785000 W:{784,1000}, b:{1,1000}
11:17:05 layer1 (OutputLayer) 1000,10 10010 W:{1000,10}, b:{1,10}
11:17:05 -------------------------------------------------------------------------
11:17:05 Total Parameters: 795010
11:17:05 Trainable Parameters: 795010
11:17:05 Frozen Parameters: 0
11:17:05 =========================================================================
[<--- snipped --->]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment