digits 4.0 0.14.0-rc.3 /Ubuntu (aws)

training eines 5-Klassen-GoogLenet-Modells mit ca. 800 Trainingsmustern in jeder Klasse. Ich habe versucht, den bvlc_imagent als vortrainiertes Modell zu verwenden. Dies sind die Schritte, die ich unternommen habe:

downloaded imagenet fromhttp: //dl.caffe.berkeleyvision.org/bvlc_googlenet.caffemode und legte es in / home / ubuntu / models

2.

ein. Fügte den "train_val.prototxt" von hier einhttps: //github.com/BVLC/caffe/blob/master/models/bvlc_reference_caffenet/train_val.prototx in die benutzerdefinierte Netzwerkregisterkarte und

b. '#' hat die Zeilen "source" und "backend" auskommentiert (da es sich um die entsprechenden Zeilen handelte)

In das Textfeld für vortrainierte Modelle wurde der Pfad zum '.caffemodel' eingefügt. in meinem Fall: "/home/ubuntu/models/bvlc_googlenet.caffemodel"

Ich erhalte diesen Fehler:

ERROR: Cannot copy param 0 weights from layer 'loss1/classifier'; shape mismatch. Source param shape is 1 1 1000 1024 (1024000); target param shape is 6 1024 (6144). To learn this layer's parameters from scratch rather than copying from a saved net, rename the layer.

Ich habe verschiedene train_val.prototext aus Github-Fragen usw. eingefügt und leider kein Glück,

Ich bin mir nicht sicher, warum dies so kompliziert wird. In älteren Ziffernversionen konnten wir einfach den Pfad zum Ordner eingeben und es funktionierte hervorragend für das Transferlernen.

Kann jemand helfen?