Method size

lesson5-TrainingVGG/dataSet.h:33–35

Return the length of data

    }
    // Return the length of data
    torch::optional<size_t> size() const override {
        return image_paths.size();
    }
private:
    std::vector<std::string> image_paths;
    std::vector<int> labels;