From 489207111f1cdf0d7e3bcfc2410a92121c58f3cd Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Fri, 11 Aug 2017 11:51:39 +0000
Subject: [PATCH] Some fixes
---
src/parser.c | 10 +++++++++-
1 files changed, 9 insertions(+), 1 deletions(-)
diff --git a/src/parser.c b/src/parser.c
index 3f39a13..922227a 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;
@@ -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