TorchCraftAI
A bot for machine learning research on StarCraft: Brood War
|
This is the complete list of members for cpid::ESTrainer, including all inherited members.
activeMapMutex_ | cpid::Trainer | protected |
actives_ | cpid::Trainer | protected |
allGenerations_ | cpid::ESTrainer | protected |
allRewards_ | cpid::ESTrainer | protected |
allSeeds_ | cpid::ESTrainer | protected |
antithetic_ | cpid::ESTrainer | protected |
batchBarrier_ | cpid::ESTrainer | protected |
batcher_ | cpid::Trainer | protected |
batchSize_ | cpid::ESTrainer | protected |
cast(ReplayBuffer::Episode const &e) | cpid::Trainer | inlineprotected |
currentModelMutex_ | cpid::ESTrainer | protected |
done_ | cpid::Trainer | protected |
epGuard_ | cpid::Trainer | protected |
ESTrainer(ag::Container model, ag::Optimizer optim, std::unique_ptr< BaseSampler > sampler, float std, size_t batchSize, size_t historyLength, bool antithetic, RewardTransform transform, bool onPolicy) | cpid::ESTrainer | |
evaluatorFactory(ag::Container model, std::unique_ptr< BaseSampler > s, size_t n, ForwardFunction func) | cpid::Trainer | protectedstatic |
forceStopEpisode(EpisodeHandle const &) override | cpid::ESTrainer | virtual |
forward(ag::Variant inp, EpisodeHandle const &) override | cpid::ESTrainer | virtual |
ForwardFunction typedef | cpid::Trainer | protected |
forwardUnbatched(ag::Variant in, ag::Container model=nullptr) | cpid::Trainer | |
gamesStarted_ | cpid::ESTrainer | protected |
gameToGenerationSeed_ | cpid::ESTrainer | protected |
gatherSize_ | cpid::ESTrainer | protected |
generateModel(int generation, int64_t seed) | cpid::ESTrainer | protected |
generations_ | cpid::ESTrainer | protected |
getGameModel(GameUID const &gameIUID, EpisodeKey const &key) | cpid::ESTrainer | |
historyLength_ | cpid::ESTrainer | protected |
insertionMutex_ | cpid::ESTrainer | protected |
is() const | cpid::Trainer | inline |
isActive(EpisodeHandle const &) | cpid::Trainer | |
isDone() const | cpid::Trainer | inline |
isTrain() const | cpid::Trainer | inline |
kFwdMetricsSubsampling | cpid::Trainer | protectedstatic |
kNone enum value | cpid::ESTrainer | |
kRankTransform enum value | cpid::ESTrainer | |
kStdNormalize enum value | cpid::ESTrainer | |
load(Archive &ar) | cpid::Trainer | inline |
makeEvaluator(size_t n, std::unique_ptr< BaseSampler > sampler=std::make_unique< DiscreteMaxSampler >()) override | cpid::ESTrainer | virtual |
makeFrame(ag::Variant trainerOutput, ag::Variant state, float reward) override | cpid::ESTrainer | virtual |
metricsContext() const | cpid::Trainer | inline |
metricsContext_ | cpid::Trainer | protected |
model() const | cpid::Trainer | |
model_ | cpid::Trainer | protected |
modelCache_ | cpid::ESTrainer | protected |
modelsHistory_ | cpid::ESTrainer | protected |
modelStorageMutex_ | cpid::ESTrainer | protected |
modelWriteMutex_ | cpid::Trainer | protected |
newGames_ | cpid::ESTrainer | protected |
onPolicy_ | cpid::ESTrainer | protected |
optim() const | cpid::Trainer | |
optim_ | cpid::Trainer | protected |
populateSeedQueue() | cpid::ESTrainer | protected |
replayBuffer() | cpid::Trainer | |
replayer_ | cpid::Trainer | protected |
reset() override | cpid::ESTrainer | virtual |
rewards_ | cpid::ESTrainer | protected |
RewardTransform enum name | cpid::ESTrainer | |
rewardTransform(torch::Tensor const &rewards, RewardTransform transform) | cpid::ESTrainer | |
sample(ag::Variant in) | cpid::Trainer | |
sampler_ | cpid::Trainer | protected |
save(Archive &ar) const | cpid::Trainer | inline |
seedQueue_ | cpid::ESTrainer | protected |
seedQueueMutex_ | cpid::ESTrainer | protected |
seeds_ | cpid::ESTrainer | 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::ESTrainer | virtual |
std_ | cpid::ESTrainer | protected |
step(EpisodeHandle const &, std::shared_ptr< ReplayBufferFrame > v, bool isDone=false) | cpid::Trainer | virtual |
stepEpisode(GameUID const &, EpisodeKey const &, ReplayBuffer::Episode &) override | cpid::ESTrainer | protectedvirtual |
stepFrame(GameUID const &, EpisodeKey const &, ReplayBuffer::Episode &) | cpid::Trainer | inlineprotectedvirtual |
stepGame(GameUID const &game) | cpid::Trainer | inlineprotectedvirtual |
TORCH_ARG(bool, waitUpdate) | cpid::ESTrainer | |
cpid::Trainer::TORCH_ARG(float, noiseStd) | cpid::Trainer | |
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 | |
transform_ | cpid::ESTrainer | protected |
update() override | cpid::ESTrainer | virtual |
updateMutex_ | cpid::ESTrainer | protected |
~Trainer()=default | cpid::Trainer | virtual |