I use python (The IDE is Pycharm) to run a net and a pretrained model to get 8 outputs. The code is like this:
myNet = root + netName
myModel = root + modelName
net = caffe.Net(myNet , myModel , caffe.TEST)
The net is really simple:
name: "Trial"
layer {
name: "data"
type: "MemoryData"
top: "imgPairsSubMean"
top: "boundings"
memory_data_param {
batch_size: 1
channels: 2
height: 64
width: 64
}
include {
phase: TEST
}
}
layer {
name: "conv1"
type: "Convolution"
bottom: "imgPairsSubMean"
top: "conv1"
param {
lr_mult: 1
}
param {
lr_mult: 2
}
convolution_param {
num_output: 1
pad: 1
kernel_size: 3
stride: 1
weight_filler {
type: "xavier"
std: 0.01
}
bias_filler {
type: "constant"
value: 0
}
}
}
layer {
name: "relu1"
type: "ReLU"
bottom: "conv1"
top: "conv1"
}
layer {
name: "pool1"
type: "Pooling"
bottom: "conv1"
top: "pool1"
pooling_param {
pool: MAX
kernel_size: 2
stride: 2
}
}
layer {
name: "ip1"
type: "InnerProduct"
bottom: "pool1"
top: "ip1"
param {
lr_mult: 1
}
param {
lr_mult: 2
}
inner_product_param {
num_output: 8
weight_filler {
type: "gaussian"
std: 0.005
}
bias_filler {
type: "constant"
value: 0
}
}
}
The model is trained by myself. Used data size is 70000 * 2 * 64 * 64. The network can do nothing but a trial to test why I always get an error like this.
I0830 18:49:22.437019 10536 layer_factory.hpp:77] Creating layer data
I0830 18:49:22.437019 10536 net.cpp:91] Creating Layer data
I0830 18:49:22.437019 10536 net.cpp:399] data -> imgPairsSubMean
I0830 18:49:22.437019 10536 net.cpp:399] data -> boundings
I0830 18:49:22.437019 10536 net.cpp:141] Setting up data
I0830 18:49:22.437019 10536 net.cpp:148] Top shape: 1 2 64 64 (8192)
I0830 18:49:22.437019 10536 net.cpp:148] Top shape: 1 (1)
I0830 18:49:22.437019 10536 net.cpp:156] Memory required for data: 32772
I0830 18:49:22.437019 10536 layer_factory.hpp:77] Creating layer conv1
......
I0830 18:49:22.681442 10536 net.cpp:425] ip1 <- pool1
I0830 18:49:22.681442 10536 net.cpp:399] ip1 -> ip1
I0830 18:49:22.681442 10536 net.cpp:141] Setting up ip1
I0830 18:49:22.681442 10536 net.cpp:148] Top shape: 1 8 (8)
I0830 18:49:22.681442 10536 net.cpp:156] Memory required for data: 69668
I0830 18:49:22.681442 10536 net.cpp:219] ip3 does not need backward computation.
I0830 18:49:22.681442 10536 net.cpp:219] pool1 does not need backward computation.
I0830 18:49:22.681442 10536 net.cpp:219] relu1 does not need backward computation.
I0830 18:4***Check failure stack trace***
Process finished with exit code -1073740791 (0xC0000409)
Please pay attention to the last two lines.
I used to think it might be the memory limitation of Pycharm, so I changed the .vmoptions files, no use; I changed the data type from "MemoryData" to "Input", no luck.
So, can anyone give a hint what's going on here? What could I do to get rid of that? Thank you so much!
User contributions licensed under CC BY-SA 3.0