MCPcopy Create free account
hub / github.com/alibaba/MNN / DropoutModule

Method DropoutModule

tools/train/source/nn/NN.cpp:43–46  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

41class DropoutModule : public Module {
42public:
43 DropoutModule(const float dropRatio) {
44 mDropRatio = dropRatio;
45 setType("Dropout");
46 }
47
48 virtual std::vector<Express::VARP> onForward(const std::vector<Express::VARP>& inputs) override {
49 Express::VARP x = inputs[0];

Callers

nothing calls this directly

Calls 1

setTypeFunction · 0.85

Tested by

no test coverage detected