From ae74d0ef31485f84e1856b4733135d2753dbb033 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Sat, 21 Oct 2017 20:56:07 +0000
Subject: [PATCH] Fixed fread() for net->seen
---
src/parser.c | 12 ++++++++++--
1 files changed, 10 insertions(+), 2 deletions(-)
diff --git a/src/parser.c b/src/parser.c
index 3f39a13..ae9e194 100644
--- a/src/parser.c
+++ b/src/parser.c
@@ -29,6 +29,7 @@
#include "shortcut_layer.h"
#include "softmax_layer.h"
#include "utils.h"
+#include <stdint.h>
typedef struct{
char *type;
@@ -835,7 +836,7 @@
fwrite(&major, sizeof(int), 1, fp);
fwrite(&minor, sizeof(int), 1, fp);
fwrite(&revision, sizeof(int), 1, fp);
- fwrite(net.seen, sizeof(int), 1, fp);
+ fwrite(net.seen, sizeof(uint64_t), 1, fp);
int i;
for(i = 0; i < net.n && i < cutoff; ++i){
@@ -1023,7 +1024,14 @@
fread(&major, sizeof(int), 1, fp);
fread(&minor, sizeof(int), 1, fp);
fread(&revision, sizeof(int), 1, fp);
- fread(net->seen, sizeof(int), 1, fp);
+ if ((major * 10 + minor) >= 2) {
+ fread(net->seen, sizeof(uint64_t), 1, fp);
+ }
+ else {
+ int iseen = 0;
+ fread(&iseen, sizeof(int), 1, fp);
+ *net->seen = iseen;
+ }
int transpose = (major > 1000) || (minor > 1000);
int i;
--
Gitblit v1.10.0