Home
last modified time | relevance | path

Searched defs:net_with_loss (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/nn/probability/infer/variational/
Dsvi.py42 def __init__(self, net_with_loss, optimizer): argument
/third_party/mindspore/tests/st/probability/bnn_layers/
Dtest_bnn_layer.py130 net_with_loss = bnn_layers.WithBNNLossCell(network, criterion, 60000, 0.000001) variable
/third_party/mindspore/tests/st/probability/transforms/
Dtest_transform_bnn_model.py130 net_with_loss = WithLossCell(network, criterion) variable
Dtest_transform_bnn_layer.py131 net_with_loss = WithLossCell(network, criterion) variable