OmniSciDB  a5dc49c757
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
ModelInfo Struct Reference

#include <TorchWrapper.h>

Public Attributes

bool is_valid {false}
 
int64_t batch_size {-1}
 
int64_t raster_channels {-1}
 
int64_t raster_tile_width {-1}
 
int64_t raster_tile_height {-1}
 
int64_t stride {-1}
 
std::vector< std::string > class_labels
 

Detailed Description

Definition at line 46 of file TorchWrapper.h.

Member Data Documentation

int64_t ModelInfo::batch_size {-1}

Definition at line 48 of file TorchWrapper.h.

Referenced by get_model_info_from_json().

std::vector<std::string> ModelInfo::class_labels
bool ModelInfo::is_valid {false}

Definition at line 47 of file TorchWrapper.h.

Referenced by __attribute__(), and get_model_info_from_json().

int64_t ModelInfo::raster_channels {-1}

Definition at line 49 of file TorchWrapper.h.

Referenced by get_model_info_from_json().

int64_t ModelInfo::raster_tile_height {-1}

Definition at line 51 of file TorchWrapper.h.

Referenced by get_model_info_from_json().

int64_t ModelInfo::raster_tile_width {-1}

Definition at line 50 of file TorchWrapper.h.

Referenced by get_model_info_from_json().

int64_t ModelInfo::stride {-1}

Definition at line 52 of file TorchWrapper.h.

Referenced by get_model_info_from_json().


The documentation for this struct was generated from the following file: