TorchCraftAI
A bot for machine learning research on StarCraft: Brood War
|
This is the complete list of members for cpid::BatchedPGTrainer, including all inherited members.
activeMapMutex_ | cpid::Trainer | protected |
actives_ | cpid::Trainer | protected |
BatchedPGTrainer(ag::Container model, ag::Optimizer optim, std::unique_ptr< BaseSampler > sampler, double gamma=0.99, int batchSize=10, std::size_t maxBatchSize=50, std::unique_ptr< AsyncBatcher > batcher=nullptr) | cpid::BatchedPGTrainer | |
batcher_ | cpid::Trainer | protected |
cast(ReplayBuffer::Episode const &e) | cpid::Trainer | inlineprotected |
done_ | cpid::Trainer | protected |
doOnlineUpdatesInstead() | cpid::BatchedPGTrainer | |
epGuard_ | cpid::Trainer | protected |
episodes() | cpid::BatchedPGTrainer | inline |
evaluatorFactory(ag::Container model, std::unique_ptr< BaseSampler > s, size_t n, ForwardFunction func) | cpid::Trainer | protectedstatic |
forceStopEpisode(EpisodeHandle const &) | cpid::Trainer | virtual |
forward(ag::Variant inp, EpisodeHandle const &) override | cpid::BatchedPGTrainer | virtual |
ForwardFunction typedef | cpid::Trainer | protected |
forwardUnbatched(ag::Variant in, ag::Container model=nullptr) | cpid::Trainer | |
is() const | cpid::Trainer | inline |
isActive(EpisodeHandle const &) | cpid::Trainer | |
isDone() const | cpid::Trainer | inline |
isTrain() const | cpid::Trainer | inline |
kFwdMetricsSubsampling | cpid::Trainer | protectedstatic |
load(Archive &ar) | cpid::Trainer | inline |
makeEvaluator(size_t, std::unique_ptr< BaseSampler > sampler=std::make_unique< DiscreteMaxSampler >()) override | cpid::BatchedPGTrainer | virtual |
makeFrame(ag::Variant trainerOutput, ag::Variant state, float reward) override | cpid::BatchedPGTrainer | virtual |
metricsContext() const | cpid::Trainer | inline |
metricsContext_ | cpid::Trainer | protected |
model() const | cpid::Trainer | |
model_ | cpid::Trainer | protected |
modelWriteMutex_ | cpid::Trainer | protected |
optim() const | cpid::Trainer | |
optim_ | cpid::Trainer | protected |
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 |
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() | cpid::Trainer | virtual |
step(EpisodeHandle const &, std::shared_ptr< ReplayBufferFrame > v, bool isDone=false) | cpid::Trainer | virtual |
stepEpisode(GameUID const &, EpisodeKey const &, ReplayBuffer::Episode &) override | cpid::BatchedPGTrainer | protectedvirtual |
stepFrame(GameUID const &, EpisodeKey const &, ReplayBuffer::Episode &) | cpid::Trainer | inlineprotectedvirtual |
stepGame(GameUID const &game) | cpid::Trainer | inlineprotectedvirtual |
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::BatchedPGTrainer | virtual |
~Trainer()=default | cpid::Trainer | virtual |