PyTorch'ta Önceden Eğitilmiş Bir Model Nasıl İçe Aktarılır?

Pytorch Ta Onceden Egitilmis Bir Model Nasil Ice Aktarilir



PyTorch'taki Makine Öğrenimi modelleri, milyonlarca satır ve terabaytlarca veriyle son derece karmaşık ve ayrıntılı olabilir. Eğitimde kullanılan veri kümesi ne kadar büyük ve çeşitli olursa modelin çıkarımları da o kadar iyi olur. Yeni verilerden çıkarımlar yapmak için önceden eğitilmiş modelleri kullanabilmek hayati önem taşıyor çünkü kaynakları koruyabilir ve aynı titizlikle hazırlanmış modelleri kullanabilir.

Bu blogda, önceden eğitilmiş bir modelin PyTorch'a nasıl aktarılacağına dair iki yöntemi ele alacağız.

Torchvision Kullanarak PyTorch'a Önceden Eğitilmiş Bir Model Nasıl İçe Aktarılır?

meşale görüşü ” kütüphanesi PyTorch'ta önceden eğitilmiş modelleri içe aktarmak için kullanılabilir. Birincil bölümün bir alt bölümüdür meşale ” kütüphanesidir ve daha önce derlenmiş veri kümelerinin ve eğitilmiş modellerin işlevselliğini içerir. Bu kütüphane kullanıcılara geniş bir veri seti üzerinde eğitilmiş modelleri çağırma olanağı sağlar. Önceden eğitilmiş bu modeller yeni verilere uygulanabilir ve uzun ve yönetilmesi zor eğitim döngülerine gerek kalmadan geçerli çıkarımlar sağlayabilir.







Torchvision kullanarak PyTorch'ta önceden eğitilmiş bir modelin nasıl içe aktarılacağını öğrenmek için aşağıdaki adımları izleyin:



1. Adım: Google Colab'ı açın
İşbirliğine git İnternet sitesi Google tarafından oluşturuldu ve bir ' Yeni Defter ” projeye başlamak için:







Adım 2: Gerekli Kitaplıkları İçe Aktarın
Colab IDE kurulduktan sonra ilk adım, projede ihtiyaç duyulan kitaplıkları yükleyip içe aktarmaktır:

! pip kurulum meşalesi

içe aktarmak meşale
içe aktarmak meşale görüşü
içe aktarmak meşale görüşü. modeller

Yukarıdaki kod şu şekilde çalışır:



  • pip 'Python için paket yükleyicisi' yüklemek için kullanılır meşale ' kütüphane.
  • Daha sonra “ içe aktarmak ” komutu kütüphaneyi Colab projesine aktarmak için kullanılır.
  • Sonra ' meşale görüşü ” kütüphanesi projeye aktarıldı. Bu, veri kümeleri ve modeller için işlevsellik içerir.
  • torchvision.model 'modül, Artık Sinir Ağı modelleri gibi önceden eğitilmiş çeşitli modelleri içerir' ResNet ”:

3. Adım: Önceden Eğitilmiş Modeli İçe Aktarın
Aşağıdaki kod satırını kullanarak 'torchvision.models' paketinde kayıtlı, önceden eğitilmiş bir modeli içe aktarın:

Pre_Trained_Model = meşale görüşü. modeller . ciddi50 ( önceden eğitilmiş = Doğru )

Yukarıdaki kod satırı şu şekilde çalışır:

  • Bir değişken tanımlayın ve ona referans olarak uygun bir ad verin: “Ön_Eğitimli_Model” .
  • Kullan “torchvision.modeller” “eklemek için modül ResNet ”modeli.
  • Ekle ' ciddi50 ” modelini seçin ve “ önceden eğitilmiş=Doğru argümanı olarak:

Daha sonra “print()” yöntemini kullanarak önceden eğitilmiş modeli bir çıktı olarak görüntüleyin:

Yazdır ( Pre_Trained_Model )

Not : Torchvision kullanılarak önceden eğitilmiş bir PyTorch modelinin içe aktarılmasının ayrıntılarını içeren Colab not defterimize buradan erişebilirsiniz. bağlantı .

Hugging Face Veritabanından Önceden Eğitilmiş Bir PyTorch Modeli Nasıl İçe Aktarılır?

Önceden eğitilmiş bir modeli içe aktarmanın başka bir yöntemi de onu Hugging Face platformundan elde etmektir. Hugging Face, veri bilimcileri ve programcılar için önceden eğitilmiş modeller ve büyük veri kümeleri içeren en popüler çevrimiçi veritabanlarından biridir.

Hugging Face Veri Kümesinden önceden eğitilmiş bir PyTorch modelini içe aktarmak için aşağıdaki adımları izleyin:

1. Adım: Colab Notebook'u Başlatın ve Gerekli Kitaplıkları Kurun ve İçe Aktarın
İlk adım, Colab IDE'de bir not defteri başlatmak ve ' pip ” paket yükleyicisini seçin ve bunları “ kullanarak içe aktarın içe aktarmak ' emretmek:

! pip kurulum meşalesi
! pip kurulum transformatörleri

içe aktarmak meşale
içe aktarmak transformatörler
transformatörlerden içe aktarmak Otomatik Model

Bu projede aşağıdaki kütüphanelere ihtiyaç var

  • meşale ” kütüphanesi temel PyTorch kütüphanesidir.
  • transformatörler ” kitaplığı Hugging Face'in işlevlerini, modellerini ve veri kümelerini içerir:

Adım 2: Modeli Hugging Face'ten İçe Aktarın
Bu örnekte içe aktarılacak model “ Sarılma Yüz ' veritabanı burada mevcut bağlantı . Kullan ' AutoModel.from_pretrained() ” Hugging Face'ten önceden eğitilmiş bir modeli aşağıda gösterildiği gibi içe aktarma yöntemi:

pre_trained_model_name = 'Helsinki-NLP/opus-mt-en-zh'
pre_trained_model = AutoModel. from_pretrained ( pre_trained_model_name )

Yazdır ( pre_trained_model )

Yukarıdaki kod şu şekilde çalışır:

  • Model adını Hugging Face platformundaki web sitesinden kopyalayın ve ' pre_trained_model_name ” Colab'daki değişken.
  • Daha sonra “ AutoModel.from_pretrained() ” yöntemini kullanın ve model adı değişkenini argüman olarak girin.
  • Son olarak şunu kullanın: 'Yazdır() ” İçe aktarılan modeli çıktıda sergileme yöntemini kullanın.

Hugging Face'ten içe aktarılan önceden eğitilmiş model aşağıdaki çıktıyı gösterecektir:

Not : Hugging Face'ten önceden eğitilmiş bir modelin nasıl içe aktarılacağının ayrıntılarını içeren Colab not defterimize buradan ulaşabilirsiniz. bağlantı .

Profesyonel İpucu

Hugging Face, derin öğrenme projelerinde herkesin ücretsiz olarak kullanabileceği, büyük veri kümelerinden ve karmaşık modellerden oluşan değerli bir koleksiyondur. Ayrıca başkalarının kullanması için kendi veri kümelerinizi de yükleyebilirsiniz ve platform, tüm dünyadaki veri bilimcileri ve geliştiriciler arasındaki işbirliğine göre ayarlanmıştır.

Başarı! Torchvision kütüphanesini kullanarak veya transformatör kütüphanesini kullanarak Hugging Face veritabanından önceden eğitilmiş bir PyTorch modelinin nasıl içe aktarılacağını gösterdik.

Çözüm

PyTorch'ta önceden eğitilmiş bir modeli içe aktarmak için kullanıcılar torchvision kitaplığını kullanabilir veya Google Colab'daki transformatör kitaplığını kullanarak Hugging Face çevrimiçi veritabanından yararlanabilir. Bu önceden eğitilmiş modeller, eğitimde değerli zaman ve donanım kaynaklarının harcanmasını önlemek ve güvenilir çıkarımlar için doğrudan yeni verileri test etmeye gitmek için kullanılır. Bu blogda, önceden eğitilmiş modelleri PyTorch'a içe aktarmak için iki yöntem gösterdik.