... labels=labels) loss = outputs[0] # backward loss.backward() optimizer.step() # 紀錄當前batch loss running_loss += loss.item() # 計算分類 ...
確定！ 回上一頁
