Show layer type in name.

2018-03-26 15:46:36 +02:00
parent e62ab159a8
commit 0f699c1881
3 changed files with 19 additions and 9 deletions
--- a/src/fmri/LayerInfo.hpp
+++ b/src/fmri/LayerInfo.hpp
@@ -32,8 +32,7 @@ namespace fmri
        const std::vector<boost::shared_ptr<caffe::Blob<DType>>>& parameters() const;

        static Type typeByName(std::string_view name);
-
-        friend std::ostream& operator<<(std::ostream& out, Type type);
+        static std::string_view nameByType(Type type);

    private:
        std::vector<boost::shared_ptr<caffe::Blob<DType>>> parameters_;