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/network.h | 3 ++-
src/parser.c | 2 +-
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/src/network.h b/src/network.h
index bb65eac..693105c 100644
--- a/src/network.h
+++ b/src/network.h
@@ -2,6 +2,7 @@
#ifndef NETWORK_H
#define NETWORK_H
+#include <stdint.h>
#include "layer.h"
#ifdef __cplusplus
@@ -20,7 +21,7 @@
float *workspace;
int n;
int batch;
- int *seen;
+ uint64_t *seen;
float epoch;
int subdivisions;
float momentum;
diff --git a/src/parser.c b/src/parser.c
index 922227a..ae9e194 100644
--- a/src/parser.c
+++ b/src/parser.c
@@ -836,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){
--
Gitblit v1.10.0