chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,17 @@
|
||||
//
|
||||
// Dataset.cpp
|
||||
// MNN
|
||||
//
|
||||
// Created by MNN on 2020/02/17.
|
||||
// Copyright © 2018, Alibaba Group Holding Limited
|
||||
//
|
||||
|
||||
#include "Dataset.hpp"
|
||||
namespace MNN {
|
||||
namespace Train {
|
||||
|
||||
DataLoader* DatasetPtr::createLoader(const int batchSize, const bool stack, const bool shuffle, const int numWorkers) {
|
||||
return DataLoader::makeDataLoader(mDataset, batchSize, stack, shuffle, numWorkers);
|
||||
}
|
||||
} // namespace Train
|
||||
} // namespace MNN
|
||||
Reference in New Issue
Block a user