| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/ |
| D | train_and_test_multinpu_ci_data_parallel.py | 78 ds_train = create_dataset(data_path, train_mode=True, epochs=1,
82 print("ds_train.size: {}".format(ds_train.get_dataset_size()))
96 …ckptconfig = CheckpointConfig(save_checkpoint_steps=ds_train.get_dataset_size(), keep_checkpoint_m…
101 model.train(epochs, ds_train,
102 … callbacks=[TimeMonitor(ds_train.get_dataset_size()), eval_callback, callback, ckpoint_cb])
|
| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/ |
| D | train_and_test_multinpu_ci.py | 80 ds_train = create_dataset(data_path, train_mode=True, epochs=1, batch_size=batch_size,
84 print("ds_train.size: {}".format(ds_train.get_dataset_size()))
99 model.train(epochs, ds_train,
100 callbacks=[TimeMonitor(ds_train.get_dataset_size()), eval_callback, callback])
|
| /third_party/mindspore/tests/st/broadcast/ |
| D | lenet_broadcast_auto_parallel.py | 51 … ds_train = create_dataset(os.path.join('/home/workspace/mindspore_dataset/mnist', "train"), 32, 1)
56 time_cb = TimeMonitor(data_size=ds_train.get_dataset_size())
61 model.train(1, ds_train, callbacks=[time_cb, LossMonitor()])
|
| /third_party/mindspore/tests/st/model_zoo_tests/DeepFM/ |
| D | test_deepfm.py | 44 ds_train = create_dataset(dataset_path,
58 time_callback = TimeMonitor(data_size=ds_train.get_dataset_size())
72 model.train(train_config.train_epochs, ds_train, callbacks=callback_list)
|
| /third_party/mindspore/tests/st/probability/dpn/ |
| D | test_gpu_svi_vae.py | 102 ds_train = create_dataset(image_path, 128, 1)
107 vae = vi.run(train_dataset=ds_train, epochs=5)
113 for sample in ds_train.create_dict_iterator(output_numpy=True, num_epochs=1):
|
| D | test_gpu_svi_cvae.py | 112 ds_train = create_dataset(image_path, 128, 1)
118 cvae = vi.run(train_dataset=ds_train, epochs=5)
125 for sample in ds_train.create_dict_iterator(output_numpy=True, num_epochs=1):
|
| D | test_gpu_vae_gan.py | 162 ds_train = create_dataset(image_path, 128, 1)
165 vae_gan = vi.run(train_dataset=ds_train, epochs=5)
|
| /third_party/mindspore/tests/st/summary/ |
| D | test_summary_ops.py | 100 ds_train = create_mnist_dataset('train', num_samples=1, batch_size=1)
101 ds_train_iter = ds_train.create_dict_iterator()
108 model.train(1, ds_train, dataset_sink_mode=False)
|
| D | test_summary_collector.py | 118 ds_train = create_mnist_dataset("train", num_samples=num_samples)
119 model.train(1, ds_train, callbacks=[summary_collector], dataset_sink_mode=dataset_sink_mode)
|
| /third_party/mindspore/tests/st/quantization/lenet_quant/ |
| D | test_lenet_quant.py | 43 ds_train = create_dataset(os.path.join(data_path, "train"), cfg.batch_size, 1)
44 step_size = ds_train.get_dataset_size()
85 model.train(cfg['epoch_size'], ds_train, callbacks=[ckpt_callback, LossMonitor()],
|
| /third_party/mindspore/tests/st/probability/zhusuan/vae/ |
| D | vae_mnist.py | 143 ds_train = create_dataset(os.path.join(mnist_path, "train"), batch_size, repeat_size)
144 model.train(epoch_size, ds_train, callbacks=[LossMonitor()], dataset_sink_mode=False)
148 iterator = ds_train.create_tuple_iterator()
|
| /third_party/mindspore/tests/st/profiler/ |
| D | test_profiler.py | 179 ds_train = create_dataset(os.path.join(self.mnist_path, "train"))
180 if ds_train.get_dataset_size() == 0:
188 model.train(1, ds_train, dataset_sink_mode=True)
|
| /third_party/mindspore/tests/st/probability/toolbox/ |
| D | test_uncertainty.py | 122 ds_train = create_dataset('workspace/mnist/train') variable
125 train_dataset=ds_train,
|
| /third_party/mindspore/tests/st/ps/full_ps/ |
| D | test_full_ps_lenet.py | 131 ds_train = create_dataset(os.path.join(dataset_path, "train"), 32, 1) variable
132 model.train(1, ds_train, callbacks=[LossMonitor()], dataset_sink_mode=False)
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_auto_parallel_double_subgraphs.py | 163 ds_train = DatasetLenet(Tensor(batch_ids), None)
165 model.train(1, ds_train, dataset_sink_mode=False)
|
| /third_party/mindspore/tests/st/networks/ |
| D | test_gradient_accumulation.py | 212 ds_train = create_dataset(os.path.join("/home/workspace/mindspore_dataset/mnist", "train"), 32)
220 model.train_process(2, ds_train, mini_steps=4)
|
| D | test_gpu_lenet.py | 201 … ds_train = create_dataset(os.path.join('/home/workspace/mindspore_dataset/mnist', "train"), 32, 1)
202 model.train(1, ds_train, callbacks=[LossMonitor()], dataset_sink_mode=True)
|
| /third_party/mindspore/mindspore/nn/probability/ |
| D | README.md | 237 vae = vi.run(train_dataset=ds_train, epochs=10)
246 for sample in ds_train.create_dict_iterator():
450 ds_train = create_dataset('workspace/mnist/train')
453 train_dataset=ds_train,
|