Home
last modified time | relevance | path

Searched refs:in_drop_ratio (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/tests/st/gnn/
Daggregator.py257 in_drop_ratio=0.0, argument
265 self.in_drop_ratio = in_drop_ratio
266 self.in_drop = nn.Dropout(keep_prob=1 - in_drop_ratio)
267 self.in_drop_2 = nn.Dropout(keep_prob=1 - in_drop_ratio)
372 in_drop_ratio=in_drop,
Dtest_gnn_aggregator.py64 in_drop_ratio=0.6,