pytorch一點點注意事項

2021-09-24 01:19:13 字數 911 閱讀 5913

多工學習:分類+回歸

dataloader:

def __getitem__(self, idx):

image, cate, valence, arousal = self.image_list[idx].rstrip().split(' ')

img_name = os.path.join(self.root_dir,image)

image = image.open(img_name);

label = int(cate)

''' 此處請注意,pytorch支援float32, '''

valence = np.float32(valence)

arousal = np.float32(arousal)

if self.transform:

image = self.transform(image)

# sample =

''' #此處不能用list()如:[valence, arousal],要用,np.array()。 '''

sample =

model:

''' #兩個離散值就當做2分類的任務,但loss用mse妥妥的 '''

self.fc_va = nn.linear(512 * block.expansion, 2)

v_a = self.fc_va(x)

x = self.fc(x)

return x, v_a

q:一跑pytorch,未使用多gpu,且電腦卡主

r:忘使用下面這段話,導致在cpu上執行大程式,導致了接下來的結果。

model = torch.nn.dataparallel(model).cuda()

一點點感想

寫在這喧囂與寂寞的世界 今天我與chen在風初停的早上跑去借書,借了安妮的 清醒記 兩本三毛的書和一本yilan的旅行與味蕾的書。從圖書館出來,風似乎有些冷,想到又要到那陰冷的梅花館坐在又涼又硬的椅子上上一節乏味的數學課我們便都不約而同的想到了逃課。帶著心裡的負罪感我們躲進八食取暖。無意中我們嘗到了...

一點點感悟

這兩天一直在努力的調整狀態,考慮了很多,也想通了很多,因為看開了,所以狀態也好了很多。明白了,一件事情的兩面性,壞的一面會使自己痛苦不堪,好的一面會使自己心情愉快。看了一篇文章,感觸很大,裡面有很多啟發自己的語句。二十幾歲的人,除了愛情之外,還需要有一項能使自己生活的本領 要把自己的感觸 心情和生活...

一點點想法

最近公司要做個專案,用到進銷存系統,考查了好幾個,都不是很滿意 要麼功能上不行,要麼 上不行 突然在想,如果能用php做乙個,將是個不錯的選擇,因為進銷存大部分都是內部人使用,併發量要求並不是很高,主要是資料準確性和安全性,我想這兩方面重不在語言,另外乙個麻煩的地方是報表這塊,可以用其它語言來單獨處...