Searched defs:weight_dir (Results 1 – 2 of 2) sorted by relevance
17 def _get_embedding_weight(weight_dir, state_dict): argument223 def dump_embedding_lut_for_cmdline(weight_dir, state_dict, config): argument365 def get_embedding_layer(config, weight_dir, state_dict): argument436 def load_checkpoints(weight_dir): argument
189 def check_weights_exist(weight_dir): argument