"device_map": []
}
args = namedtuple("args_namedtuple", ",".join(common.keys()))
for k, v in common.items():
setattr(args, k, v)
for pool_layer in range(1, 13):
setattr(args, "pooling_layer", [-pool_layer])
server = BertServer(args)
server.start()
After Change
//save bert vectors and labels
stacked_subset_vec_all_layers = np.stack(subset_vec_all_layers)
np.save("example7_5k_2",stacked_subset_vec_all_layers)
np_subset_label = np.array(subset_label)
np.save("example7_5k_2_subset_label",np_subset_label)
//load bert vectors and labels