Update namespaces inside torch.utils.data to the latest. (#13167)
* Update torch.utils.data namespaces to the latest. * Format * Update Dataloader. * Style
This commit is contained in:
@@ -206,7 +206,7 @@ class TokenClassificationTask:
|
||||
if is_torch_available():
|
||||
import torch
|
||||
from torch import nn
|
||||
from torch.utils.data.dataset import Dataset
|
||||
from torch.utils.data import Dataset
|
||||
|
||||
class TokenClassificationDataset(Dataset):
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user