Merge pull request #1761 from srdg/patch-1

Fixed "Invalid F spec" error while building FullyConnected layer.
This commit is contained in:
zdenop 2018-07-06 22:11:36 +02:00 committed by GitHub
commit 71bb9c373b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -426,7 +426,7 @@ Network* NetworkBuilder::ParseFullyConnected(const StaticShape& input_shape,
tprintf("Invalid nonlinearity on F-spec!: %s\n", *str);
return nullptr;
}
int depth = strtol(*str + 1, str, 10);
int depth = strtol(*str + 2, str, 10);
if (depth <= 0) {
tprintf("Invalid F spec!:%s\n", *str);
return nullptr;