Torch學習記錄

pytorch有一些基礎函數須要瞭解,在本文章記錄一下。web torch.utils.data.DataLoader DataLoader是導入圖片的操做,裏面有一些參數,好比batch_size和shuffle等,默認load進去的圖片類型是PIL.Image.open的類型網絡 torchvision.transforms torchvision.transforms裏面的操做是對導入的圖片
相關文章
相關標籤/搜索