Skip to content

Commit

Permalink
Update yannpp. Use new convolution layer
Browse files Browse the repository at this point in the history
  • Loading branch information
Vearol committed Feb 18, 2019
1 parent 1f084a1 commit 86a9744
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion loaders/fs_loader_factory.h
Expand Up @@ -36,7 +36,7 @@ class fs_loader_factory {
public:
std::shared_ptr<layer_loader_t<T>> loader(std::shared_ptr<yannpp::layer_base_t<T>> const &layer) {
auto fc_layer = std::dynamic_pointer_cast<yannpp::fully_connected_layer_t<T>>(layer);
auto cn_layer = std::dynamic_pointer_cast<yannpp::convolution_layer_t<T>>(layer);
auto cn_layer = std::dynamic_pointer_cast<yannpp::convolution_layer_base_t<T>>(layer);
auto &layer_name = layer->get_metadata().name;
if (fc_layer != nullptr) {
return fc(layer_name);
Expand Down
2 changes: 1 addition & 1 deletion main.cpp
Expand Up @@ -39,7 +39,7 @@ static yannpp::activator_t<float> relu_activator(yannpp::relu_v<float>, yannpp::
layers_list create_layers() {
using namespace yannpp;

using cnn_t = convolution_layer_t<float>;
using cnn_t = convolution_layer_2d_t<float>;
using fc_t = fully_connected_layer_t<float>;
using pl_t = pooling_layer_t<float>;
using m_t = layer_metadata_t;
Expand Down

0 comments on commit 86a9744

Please sign in to comment.