MCPcopy Create free account
hub / github.com/Yonv1943/Python / build_mlp

Function build_mlp

Demo_deep_learning/DEMO_DistDataParallel.py:43–58  ·  view source on GitHub ↗

build MLP (MultiLayer Perceptron) dims: the middle dimension, `dims[-1]` is the output dimension of this network activation: the activation function if_remove_out_layer: if remove the activation function of the output layer.

(dims: [int], activation: nn = None, if_raw_out: bool = True)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

Calls

no outgoing calls

Tested by

no test coverage detected