Слияние кода завершено, страница обновится автоматически
# Copyright (c) OpenMMLab. All rights reserved.
from torch.utils.data import random_split
def split_dataset(dataset):
dset_length = len(dataset)
first_dset_length = dset_length // 2
second_dset_length = dset_length - first_dset_length
split_tuple = (first_dset_length, second_dset_length)
first_dset, second_dset = random_split(dataset, split_tuple)
first_dset.CLASSES = dataset.CLASSES
second_dset.CLASSES = dataset.CLASSES
return [first_dset, second_dset]
Вы можете оставить комментарий после Вход в систему
Неприемлемый контент может быть отображен здесь и не будет показан на странице. Вы можете проверить и изменить его с помощью соответствующей функции редактирования.
Если вы подтверждаете, что содержание не содержит непристойной лексики/перенаправления на рекламу/насилия/вульгарной порнографии/нарушений/пиратства/ложного/незначительного или незаконного контента, связанного с национальными законами и предписаниями, вы можете нажать «Отправить» для подачи апелляции, и мы обработаем ее как можно скорее.
Комментарий ( 0 )