ff365f8ce044ef9d4af82c5044746b2f31a529c9,ludwig/collect.py,,save_tensors,#Any#Any#,153

Before Change

def save_tensors(collected_tensors, experiment_dir_name):
    for tensor_name, tensor_values in collected_tensors.items():
        np_filename = os.path.join(
            experiment_dir_name,
            make_safe_filename(tensor_name) + ".npy"

After Change

def save_tensors(collected_tensors, experiment_dir_name):
    filenames = []
    for tensor_name, tensor_value in collected_tensors:
        np_filename = os.path.join(
            experiment_dir_name,
            make_safe_filename(tensor_name) + ".npy"
        )
        np.save(np_filename, tensor_value.numpy())
        filenames.append(np_filename)
    return filenames


def print_weight_names(

Instances