reid from https://github.com/michuanhaohao/reid-strong-baseline
zhangmeng
2020-01-20 455a7bb3e42582a62a02d7baf1b1d4495bf6107c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
#pragma once
 
#include <cstddef>
#include <torch/csrc/WindowsTorchApiMacro.h>
 
namespace torch {
namespace data {
namespace samplers {
/// A base class for custom index types.
struct TORCH_API CustomBatchRequest {
  virtual ~CustomBatchRequest() = default;
 
  /// The number of elements accessed by this index.
  virtual size_t size() const = 0;
};
} // namespace samplers
} // namespace data
} // namespace torch