chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,30 @@
|
||||
//
|
||||
// DataLoaderConfig.hpp
|
||||
// MNN
|
||||
//
|
||||
// Created by MNN on 2019/11/15.
|
||||
// Copyright © 2018, Alibaba Group Holding Limited
|
||||
//
|
||||
|
||||
#ifndef DataLoaderConfig_hpp
|
||||
#define DataLoaderConfig_hpp
|
||||
#include <MNN/MNNDefine.h>
|
||||
namespace MNN {
|
||||
namespace Train {
|
||||
|
||||
class MNN_PUBLIC DataLoaderConfig {
|
||||
public:
|
||||
DataLoaderConfig() = default;
|
||||
DataLoaderConfig(size_t batchSize, size_t nWorkers = 0) : batchSize(batchSize), numWorkers(nWorkers) {
|
||||
}
|
||||
|
||||
size_t batchSize = 1;
|
||||
size_t numWorkers = 0;
|
||||
size_t numJobs = numWorkers * 2;
|
||||
bool dropLast = false;
|
||||
};
|
||||
|
||||
} // namespace Train
|
||||
} // namespace MNN
|
||||
|
||||
#endif // DataLoaderConfig
|
||||
Reference in New Issue
Block a user