reid from https://github.com/michuanhaohao/reid-strong-baseline
zhangmeng
2020-01-14 5459ba1d3f7f944aa97923ed9c09a5dbc7663928
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
#ifndef THCP_SERIALIZATION_INC
#define THCP_SERIALIZATION_INC
 
#include <torch/csrc/cuda/override_macros.h>
 
#define THC_GENERIC_FILE "torch/csrc/generic/serialization.h"
#include <THC/THCGenerateAllTypes.h>
 
#define THC_GENERIC_FILE "torch/csrc/generic/serialization.h"
#include <THC/THCGenerateBoolType.h>
 
#define THC_GENERIC_FILE "torch/csrc/generic/serialization.h"
#include <THC/THCGenerateBFloat16Type.h>
 
#endif