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