عضو کانال تلگرام ما شوید.

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

رفع مشکل Size mismatch between tensors در پای تورچ
رفع مشکل 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])

 

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

موفق باشید.

برای امتیاز به این نوشته کلیک کنید!
[کل: 1 میانگین: 5]