Learning backend cuda
Compare changes
Files
18@@ -35,6 +35,9 @@ private:
@@ -35,6 +35,9 @@ private:
@@ -67,7 +70,7 @@ public:
@@ -67,7 +70,7 @@ public:
DataProvider(const Database& database, const std::size_t batchSize, const bool shuffle = false, const bool dropLast = false);