TorchCraftAI
A bot for machine learning research on StarCraft: Brood War
Typedefs | Functions
cherrypi::defogger Namespace Reference

Typedefs

using nonlin_type = std::function< torch::Tensor(torch::Tensor)>
 
using conv_builder = std::function< ag::Container(uint32_t, uint32_t, int, int, int, bool)>
 

Functions

ag::Container conv2dBuilder (uint32_t input_size, uint32_t output_size, int convsize, int stride, int padding, bool no_bias)
 
void compare (std::string name, ag::tensor_list got)
 
Convnet SimpleConvnet (conv_builder conv, nonlin_type nonlin, int convsize, int padding, int input_size, int output_size, int depth, int stride)
 
 AUTOGRAD_CONTAINER_CLASS (MapRaceFeaturize)
 
 AUTOGRAD_CONTAINER_CLASS (Convnet)
 
 AUTOGRAD_CONTAINER_CLASS (Decoder)
 
 AUTOGRAD_CONTAINER_CLASS (DefoggerModel)
 

Typedef Documentation

using cherrypi::defogger::conv_builder = typedef std::function<ag::Container(uint32_t, uint32_t, int, int, int, bool)>
using cherrypi::defogger::nonlin_type = typedef std::function<torch::Tensor(torch::Tensor)>

Function Documentation

cherrypi::defogger::AUTOGRAD_CONTAINER_CLASS ( MapRaceFeaturize  )
cherrypi::defogger::AUTOGRAD_CONTAINER_CLASS ( Convnet  )
cherrypi::defogger::AUTOGRAD_CONTAINER_CLASS ( Decoder  )
cherrypi::defogger::AUTOGRAD_CONTAINER_CLASS ( DefoggerModel  )
void cherrypi::defogger::compare ( std::string  name,
ag::tensor_list  got 
)
ag::Container cherrypi::defogger::conv2dBuilder ( uint32_t  input_size,
uint32_t  output_size,
int  convsize,
int  stride,
int  padding,
bool  no_bias 
)
Convnet cherrypi::defogger::SimpleConvnet ( conv_builder  conv,
nonlin_type  nonlin,
int  convsize,
int  padding,
int  input_size,
int  output_size,
int  depth,
int  stride 
)