Add command results to b_list
Rename tostring -> toString in models Add datasets names to b_main command help - validation
This commit is contained in:
@@ -126,4 +126,14 @@ namespace platform {
|
||||
{
|
||||
return datasets.find(name) != datasets.end();
|
||||
}
|
||||
std::string Datasets::toString() const
|
||||
{
|
||||
std::string result;
|
||||
std::string sep = "";
|
||||
for (const auto& d : datasets) {
|
||||
result += sep + d.first;
|
||||
sep = ", ";
|
||||
}
|
||||
return "{" + result + "}";
|
||||
}
|
||||
}
|
@@ -24,6 +24,7 @@ namespace platform {
|
||||
std::pair<torch::Tensor&, torch::Tensor&> getTensors(const std::string& name);
|
||||
bool isDataset(const std::string& name) const;
|
||||
void loadDataset(const std::string& name) const;
|
||||
std::string toString() const;
|
||||
};
|
||||
};
|
||||
|
||||
|
Reference in New Issue
Block a user