AlexeyAB
2017-10-19 e50489bcd7f02276798ec5c4122a1618d4cfd058
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;
@@ -826,7 +827,7 @@
    }
#endif
    fprintf(stderr, "Saving weights to %s\n", filename);
    FILE *fp = fopen(filename, "w");
    FILE *fp = fopen(filename, "wb");
    if(!fp) file_error(filename);
    int major = 0;
@@ -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;