Searched refs:server_net_cfg (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/tests/st/fl/albert/ |
| D | cloud_eval.py | 22 from src.config import eval_cfg, server_net_cfg
63 eval_data_dir, server_net_cfg.seq_length, tokenizer, eval_cfg.batch_size,
73 albert_model_cls = AlbertModelCLS(server_net_cfg)
|
| D | cloud_train.py | 25 from src.config import train_cfg, server_net_cfg
155 albert_model_cls = AlbertModelCLS(server_net_cfg)
194 input_ids = Tensor(np.zeros((train_cfg.batch_size, server_net_cfg.seq_length), np.int32))
195 … attention_mask = Tensor(np.zeros((train_cfg.batch_size, server_net_cfg.seq_length), np.int32))
196 … token_type_ids = Tensor(np.zeros((train_cfg.batch_size, server_net_cfg.seq_length), np.int32))
|
| /third_party/mindspore/tests/st/fl/albert/src/ |
| D | config.py | 62 server_net_cfg = AlbertConfig( variable
|