From: Georgi Gerganov Date: Sat, 11 May 2024 13:42:01 +0000 (+0300) Subject: mnist : clean whitespace X-Git-Tag: upstream/0.0.1642~721 X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=d3288c3331be25857099f94213c0872076b6222b;p=pkg%2Fggml%2Fsources%2Fggml mnist : clean whitespace ggml-ci --- diff --git a/examples/mnist/convert-h5-to-ggml.py b/examples/mnist/convert-h5-to-ggml.py index a4f75365..4f8d28da 100644 --- a/examples/mnist/convert-h5-to-ggml.py +++ b/examples/mnist/convert-h5-to-ggml.py @@ -18,7 +18,6 @@ import json import numpy as np import re - import torch import torch.nn as nn import torchvision.datasets as dsets @@ -45,11 +44,11 @@ fout.write(struct.pack("i", 0x67676d6c)) # magic: ggml in hex for name in list_vars.keys(): data = list_vars[name].squeeze().numpy() - print("Processing variable: " + name + " with shape: ", data.shape) + print("Processing variable: " + name + " with shape: ", data.shape) n_dims = len(data.shape); - + fout.write(struct.pack("i", n_dims)) - + data = data.astype(np.float32) for i in range(n_dims): fout.write(struct.pack("i", data.shape[n_dims - 1 - i]))