List out of index for PreProcessingNormalize.run()

davidstutz / caffe-tools

Some tools and examples for pyCaffe including LMDB I/O, custom Python layers and monitoring training error and loss.

77 stars 30 forks source link

List out of index for PreProcessingNormalize.run() #4

Open Saums opened 5 years ago

Saums commented 5 years ago

pp_out_converted = tools.pre_processing.PreProcessingOutputLMDB(lmdb_converted) pp_convert = tools.pre_processing.PreProcessingNormalize(pp_in, pp_out_converted,normalizer =512.0, batch_size=100) pp_convert.run()

Might be the batch_size definition, any rule for that?

davidstutz commented 5 years ago

Seems more like you did not define label_column properly. Did you check that your label column is correct?