-
Notifications
You must be signed in to change notification settings - Fork 1.2k
/
layer.cc
76 lines (69 loc) · 2.31 KB
/
layer.cc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
/************************************************************
*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*
*************************************************************/
#include "neuralnet/layer.h"
#include <cblas.h>
#include <glog/logging.h>
#include <math.h>
#include <cfloat>
#include "utils/factory.h"
#include "utils/singleton.h"
namespace singa {
using std::string;
Layer* Layer::Create(const LayerProto& proto) {
auto* factory = Singleton<Factory<Layer>>::Instance();
Layer* layer = nullptr;
if (proto.has_user_type())
layer = factory->Create(proto.user_type());
else
layer = factory->Create(proto.type());
return layer;
}
const std::string Layer::ToString(bool debug, int flag) {
if (!debug)
return "";
string ret = StringPrintf("Layer %10s ", name().c_str());
if ((flag & kForward) == kForward && data_.count() !=0) {
ret += StringPrintf("data norm1 %13.9f", data_.asum_data());
} else if ((flag & kBackward) == kBackward) {
if (grad_.count() != 0)
ret += StringPrintf("grad norm1 %13.9f\n", grad_.asum_data());
}
if ((flag & kTrain) == kTrain) {
for (Param* p : GetParams()) {
ret += StringPrintf(
"param id %2d, name %10s, value norm1 %13.9f, grad norm1 %13.9f\n",
p->id(), p->name().c_str(), p->data().asum_data(),
p->grad().asum_data());
}
}
return ret;
}
const std::string LossLayer::ToString(bool debug, int flag) {
std::string disp;
if (debug) {
disp = Layer::ToString(debug, flag);
} else {
disp = metric_.ToLogString();
metric_.Reset();
}
return disp;
}
} // namespace singa