TorchCraftAI
A bot for machine learning research on StarCraft: Brood War
|
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) | |
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)> |
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 | ||
) |