TorchCraftAI
A bot for machine learning research on StarCraft: Brood War
|
This is the complete list of members for cpid::CentralTrainer, including all inherited members.
activeMapMutex_ | cpid::Trainer | protected |
actives_ | cpid::Trainer | protected |
batcher_ | cpid::Trainer | protected |
bufferPool_ | cpid::CentralTrainer | protected |
cast(ReplayBuffer::Episode const &e) | cpid::Trainer | inlineprotected |
CentralTrainer(bool isServer, ag::Container model, ag::Optimizer optim, std::unique_ptr< BaseSampler > sampler, std::unique_ptr< AsyncBatcher > batcher=nullptr) | cpid::CentralTrainer | |
CentralTrainer(ag::Container model, ag::Optimizer optim, std::unique_ptr< BaseSampler > sampler, std::unique_ptr< AsyncBatcher > batcher=nullptr) | cpid::CentralTrainer | protected |
client_ | cpid::CentralTrainer | protected |
dequeueEpisodes() | cpid::CentralTrainer | protected |
dequeueEpisodes_ | cpid::CentralTrainer | protected |
done_ | cpid::Trainer | protected |
epGuard_ | cpid::Trainer | protected |
episodeClientEnqueue(EpisodeData const &) | cpid::CentralTrainer | protectedvirtual |
evaluatorFactory(ag::Container model, std::unique_ptr< BaseSampler > s, size_t n, ForwardFunction func) | cpid::Trainer | protectedstatic |
forceStopEpisode(EpisodeHandle const &) override | cpid::CentralTrainer | virtual |
forward(ag::Variant inp, EpisodeHandle const &) override | cpid::CentralTrainer | virtual |
ForwardFunction typedef | cpid::Trainer | protected |
forwardUnbatched(ag::Variant in, ag::Container model=nullptr) | cpid::Trainer | |
getMaxBatchLength() const | cpid::CentralTrainer | protectedvirtual |
getSendInterval() const | cpid::CentralTrainer | protectedvirtual |
is() const | cpid::Trainer | inline |
isActive(EpisodeHandle const &) | cpid::Trainer | |
isDone() const | cpid::Trainer | inline |
isServer() const | cpid::CentralTrainer | inline |
isTrain() const | cpid::Trainer | inline |
kFwdMetricsSubsampling | cpid::Trainer | protectedstatic |
load(Archive &ar) | cpid::Trainer | inline |
makeEvaluator(size_t, std::unique_ptr< BaseSampler > sampler) | cpid::Trainer | virtual |
makeFrame(ag::Variant trainerOutput, ag::Variant state, float reward) override | cpid::CentralTrainer | virtual |
metricsContext() const | cpid::Trainer | inline |
metricsContext_ | cpid::Trainer | protected |
model() const | cpid::Trainer | |
model_ | cpid::Trainer | protected |
modelMutex_ | cpid::CentralTrainer | protected |
modelReadLock() | cpid::CentralTrainer | |
modelWriteLock() | cpid::CentralTrainer | |
modelWriteMutex_ | cpid::Trainer | protected |
newBatches_ | cpid::CentralTrainer | protected |
newGamesMutex_ | cpid::CentralTrainer | protected |
optim() const | cpid::Trainer | |
optim_ | cpid::Trainer | protected |
receivedFrames(GameUID const &, std::string const &)=0 | cpid::CentralTrainer | protectedpure virtual |
replayBuffer() | cpid::Trainer | |
replayer_ | cpid::Trainer | protected |
reset() | cpid::Trainer | virtual |
sample(ag::Variant in) | cpid::Trainer | |
sampler_ | cpid::Trainer | protected |
save(Archive &ar) const | cpid::Trainer | inline |
serveContinuously() const | cpid::CentralTrainer | protectedvirtual |
server_ | cpid::CentralTrainer | protected |
setBatcher(std::unique_ptr< AsyncBatcher > batcher) | cpid::Trainer | |
setDone(bool=true) | cpid::Trainer | |
setMetricsContext(std::shared_ptr< MetricsContext > context) | cpid::Trainer | inline |
setTrain(bool=true) | cpid::Trainer | |
startEpisode() override | cpid::CentralTrainer | virtual |
step(EpisodeHandle const &, std::shared_ptr< ReplayBufferFrame > v, bool isDone=false) | cpid::Trainer | virtual |
stepEpisode(GameUID const &, EpisodeKey const &, ReplayBuffer::Episode &) override | cpid::CentralTrainer | virtual |
stepFrame(GameUID const &, EpisodeKey const &, ReplayBuffer::Episode &) override | cpid::CentralTrainer | virtual |
stepGame(GameUID const &game) | cpid::Trainer | inlineprotectedvirtual |
stop_ | cpid::CentralTrainer | protected |
TORCH_ARG(float, noiseStd) | cpid::Trainer | |
TORCH_ARG(bool, continuousActions) | cpid::Trainer | |
train_ | cpid::Trainer | protected |
Trainer(ag::Container model, ag::Optimizer optim, std::unique_ptr< BaseSampler >, std::unique_ptr< AsyncBatcher > batcher=nullptr) | cpid::Trainer | |
update() override | cpid::CentralTrainer | virtual |
~CentralTrainer() | cpid::CentralTrainer | virtual |
~Trainer()=default | cpid::Trainer | virtual |