Home
last modified time | relevance | path

Searched defs:epochs (Results 1 – 15 of 15) sorted by relevance

/third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/
Ddatasets.py33 def _get_tf_dataset(data_dir, train_mode=True, epochs=1, batch_size=1000, argument
69 def _get_mindrecord_dataset(directory, train_mode=True, epochs=1, batch_size=1000, argument
110 def create_dataset(data_dir, train_mode=True, epochs=1, batch_size=1000, argument
/third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/
Ddataset.py149 def _get_h5_dataset(directory, train_mode=True, epochs=1, batch_size=1000): argument
180 def _get_mindrecord_dataset(directory, train_mode=True, epochs=1, batch_size=1000, argument
221 def _get_tf_dataset(directory, train_mode=True, epochs=1, batch_size=1000, argument
269 def create_dataset(directory, train_mode=True, epochs=1, batch_size=1000, argument
/third_party/mindspore/mindspore/nn/probability/infer/variational/
Dsvi.py52 def run(self, train_dataset, epochs=10): argument
/third_party/mindspore/tests/ut/python/dataset/
Dtest_callbacks.py102 def build_test_case_1cb(epochs, steps, step_size=1, repeat=1): argument
127 def build_test_case_2cbs(epochs, steps): argument
148 def build_test_case_2maps(epochs, steps): argument
/third_party/mindspore/mindspore/lite/java/java/fl_client/src/main/java/com/mindspore/flclient/model/
DTrainModel.java80 public int trainModel(String modelPath, int epochs) { in trainModel()
107 private int trainLoop(int epochs) { in trainLoop()
/third_party/mindspore/mindspore/lite/src/cxx_api/train/
Dmodel.cc30 Status Model::Train(int epochs, std::shared_ptr<dataset::Dataset> ds, std::vector<TrainCallBack *> … in Train()
/third_party/mindspore/mindspore/nn/probability/toolbox/
Danomaly_detection.py47 def train(self, train_dataset, epochs=5): argument
Duncertainty_evaluation.py85 def __init__(self, model, train_dataset, task_type, num_classes=None, epochs=1, argument
/third_party/mindspore/mindspore/lite/src/train/
Dtrain_loop.cc38 int TrainLoop::Train(int epochs, Dataset *ds, std::vector<session::TrainLoopCallBack *> cbs, LoadDa… in Train()
/third_party/mindspore/mindspore/lite/examples/unified_api/src/
Dnet_runner.cc84 explicit Measurement(unsigned int epochs) in Measurement()
/third_party/mindspore/mindspore/lite/tools/benchmark_train/
Dnet_train.cc337 … const TrainCfg &train_cfg, int epochs) { in CreateAndRunNetworkForTrain()
404 int epochs, bool check_accuracy) { in CreateAndRunNetwork()
/third_party/mindspore/mindspore/lite/examples/train_lenet/src/
Dnet_runner.cc82 explicit Measurement(unsigned int epochs) in Measurement()
/third_party/mindspore/mindspore/lite/java/java/fl_client/src/main/java/com/mindspore/flclient/
DFLLiteClient.java59 private int epochs = 1; field in FLLiteClient
/third_party/mindspore/mindspore/lite/tools/converter/quantizer/
Dquantize_util.cc467 std::vector<int8_t> KMeans(float *data, size_t elem_count, size_t k, size_t epochs, schema::QuantPa… in KMeans()
/third_party/boost/libs/math/doc/graphs/hypergeometric_1f1/
Dplotlyjs-bundle.js1 …on"==typeof define&&define.amd)define([],t);else{("undefined"!=typeof window?window:"undefined"!=t… property