我一直在尋找張量流寫的AlexNet模型,我發現的所有代碼都是使用一些預先訓練過的權重。是否有沒有預先訓練的重量與張量流寫的AlexNet模型?
您是否有任何想法是否存在在模型執行過程中構建權重的代碼? 謝謝。
我一直在尋找張量流寫的AlexNet模型,我發現的所有代碼都是使用一些預先訓練過的權重。是否有沒有預先訓練的重量與張量流寫的AlexNet模型?
您是否有任何想法是否存在在模型執行過程中構建權重的代碼? 謝謝。
您可以在TensorFlow存儲庫的路徑tensorflow/contrib/slim/python/slim/nets/alexnet.py
中的TensorFlow中找到AlexNet模型的定義(以前是TF-Slim的示例,現在只是tf.contrib.layers
)。
你可以在這裏找到一篇不錯的文章:Finetuning AlexNet with TensorFlow 它包含github代碼的地址