如何使用PyTorch Metric Learning實現訓練數據的類別平衡:完整指南與範例

當我們使用深度學習模型進行訓練時,數據的平衡非常重要。如果某些類別的數據比其他類別多很多,模型可能會偏向於頻繁出現的類別,導致對少數類別的預測效果不佳。為了解決這個問題,我們可以使用pytorch_metric_learning這個工具來幫助我們平衡每個batch中的類別數據。 在這篇教學中,我們將逐步介紹如何使用pytorch_metric_learning中的samplers來達到這個目的。我們也會提供一個範例,讓大家更清楚地了解如何應用這個方法。

Continue Reading如何使用PyTorch Metric Learning實現訓練數據的類別平衡:完整指南與範例

Libtorch 安裝

當網路訓練好之後,若需要部屬在C plus plus 程式碼上,這時候就需要先把 pytorch訓練出的網路(.pth) 先轉成.pt格式的網路,然後再經由facebook 所開發的另一套C++ library--libtorch來引用執行,引用前需要先針對libtorch 編譯執行,底下是詳細步驟

Continue ReadingLibtorch 安裝