reid from https://github.com/michuanhaohao/reid-strong-baseline
zhangmeng
2020-01-10 ae87fbb44d1b7cb32faa69a0698d941cb5bfc4b0
1
2
3
4
#pragma once
 
#include <torch/csrc/autograd/autograd.h>
#include <torch/csrc/autograd/custom_function.h>