رفع مشکل Size mismatch between tensors در پای تورچ

رفع مشکل 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])

اگر مشکل شما با کد بالا حل نشد در بخش نظرات این نوشته آن را برای ما بنویسید.