رفع مشکل Size mismatch between tensors در پای تورچ
اگر در حین اجرای برنامههایی که در آن پای تورچ یا Pytorch با خطای Sum of input lengths does not equal the length of the input dataset! روبرو شدهاید و این خطا را از تابع random_split دریافت میکنید این نوشته روشی برای حل مشکل شما است.
رفع مشکل Size mismatch between tensors در پای تورچ
برای رفع این مشکل کافیست کد زیر را استفاده کنید تا بخش بندی که در random_split میخواهید انجام دهید به صورت غیر دستی تنظیم شود. این مشکل معمولا زمانی ایجاد میشود که شما در تابع random_split از مقدارهای عددی مستقیما استفاده میکنید:
dataset = TensorDataset(x_tensor, y_tensor) val_size = int(len(dataset)*0.2) train_size = len(dataset)- int(len(dataset)*0.2) train_dataset, val_dataset = random_split(dataset, [train_size, val_size])
اگر مشکل شما با کد بالا حل نشد در بخش نظرات این نوشته آن را برای ما بنویسید.
برای امتیاز به این نوشته کلیک کنید!
[کل: 1 میانگین: 5]
ارسال پاسخ