Home
last modified time | relevance | path

Searched refs:dropout_ratio (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/tests/st/gnn/
Daggregator.py146 dropout_ratio=None, argument
161 self.dropout_ratio = dropout_ratio
162 if self.dropout_ratio is not None:
163 self.dropout = nn.Dropout(keep_prob=self.dropout_ratio)
164 self.dropout_flag = self.dropout_ratio is not None
202 dropout_ratio=None, argument
211 dropout_ratio,
Dtest_gnn_aggregator.py46 aggregator = MeanAggregator(32, 64, activation="relu", dropout_ratio=0.5)
53 aggregator = MeanAggregator(32, 64, activation="relu", dropout_ratio=0.5)
/third_party/mindspore/mindspore/lite/examples/export_models/models/
Dmini_alexnet.py42 dropout_ratio = 0.65
44 dropout_ratio = 1.0
49 self.dropout = nn.Dropout(dropout_ratio)
/third_party/mindspore/third_party/proto/caffe/
Dcaffe.proto695 optional float dropout_ratio = 1 [default = 0.5]; // dropout ratio field
1379 optional float dropout_ratio = 12 [default = 0.5]; // dropout ratio field