Checkpointer(std::shared_ptr< Trainer > trainer) | cpid::Checkpointer | |
checkpointPath(std::string const &path) | cpid::Checkpointer | |
checkpointPath() const | cpid::Checkpointer | |
checkpointPath_ | cpid::Checkpointer | |
checkpointTrainer(std::string const &suffix="final") | cpid::Checkpointer | |
checkpointTrainer(std::shared_ptr< Trainer > trainer, std::string const &filename="trainer_final.bin") | cpid::Checkpointer | static |
FORMAT_DEFAULT enum value | cpid::Checkpointer | |
FORMAT_TORCHBOARD enum value | cpid::Checkpointer | |
getModelPath() const | cpid::Checkpointer | |
Hook typedef | cpid::Checkpointer | |
lastEpochStamp_ | cpid::Checkpointer | protected |
lastEpochUpdateNum_ | cpid::Checkpointer | protected |
MetricsSummaryFormat enum name | cpid::Checkpointer | |
onEpoch(int updateCount) | cpid::Checkpointer | protected |
onUpdate(int updateCount) | cpid::Checkpointer | protected |
plotVisdom(const std::vector< float > &values, int count) | cpid::Checkpointer | protected |
printSummary(std::unordered_map< std::string, float > means, std::unordered_map< std::string, float > mins, std::unordered_map< std::string, float > maxs) | cpid::Checkpointer | protected |
reduceMetrics(std::vector< float > &values) | cpid::Checkpointer | protected |
TORCH_ARG(int, epochLength) | cpid::Checkpointer | |
TORCH_ARG(std::shared_ptr< visdom::Visdom >, visdom) | cpid::Checkpointer | |
TORCH_ARG(std::vector< std::string >, visdomKeys) | cpid::Checkpointer | |
TORCH_ARG(bool, visdomOnEpoch) | cpid::Checkpointer | |
TORCH_ARG(int, visdomPlotFreq) | cpid::Checkpointer | |
TORCH_ARG(std::string, compareMetric) | cpid::Checkpointer | |
TORCH_ARG(bool, printMetricsSummary) | cpid::Checkpointer | |
TORCH_ARG(bool, aggregateMetrics) | cpid::Checkpointer | |
TORCH_ARG(bool, flushMetrics) | cpid::Checkpointer | |
TORCH_ARG(bool, dumpMetrics) | cpid::Checkpointer | |
TORCH_ARG(MetricsSummaryFormat, metricsSummaryFormat) | cpid::Checkpointer | |
TORCH_ARG(bool, reduceMax) | cpid::Checkpointer | |
TORCH_ARG(Hook, epochHook) | cpid::Checkpointer | |
TORCH_ARG(Hook, updateHook) | cpid::Checkpointer | |
trainer_ | cpid::Checkpointer | protected |
updateDone(int updateCount) | cpid::Checkpointer | |
visdomLines_ | cpid::Checkpointer | protected |