This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
Repositories
Datasets
Forum
实训
竞赛
大数据
应用
Register
Sign In
hummingbird
/
fastNLP
Not watched
Unwatch
Watch all
Watch but not notify
2
Star
0
Fork
0
Code
Releases
13
Wiki
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
Browse Source
fix bugs of _FDataset of paddle, jittor and oneflow
dev0.8.0
x54-729
2 years ago
parent
46b94002af
commit
1f4208b1c9
3 changed files
with
21 additions
and
0 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+6
-0
fastNLP/core/dataloaders/jittor_dataloader/fdl.py
+6
-0
fastNLP/core/dataloaders/oneflow_dataloader/fdl.py
+9
-0
fastNLP/core/dataloaders/paddle_dataloader/fdl.py
+ 6
- 0
fastNLP/core/dataloaders/jittor_dataloader/fdl.py
View File
@@ -37,6 +37,12 @@ class _JittorDataset(Dataset):
item = item.tolist()
return (item, self.dataset[item])
def __getstate__(self):
return self.__dict__
def __setstate__(self, state):
self.__dict__ = state
class JittorDataLoader:
"""
+ 6
- 0
fastNLP/core/dataloaders/oneflow_dataloader/fdl.py
View File
@@ -47,6 +47,12 @@ class _FDataSet:
def __len__(self) -> int:
return len(self.dataset)
def __getstate__(self):
return self.__dict__
def __setstate__(self, state):
self.__dict__ = state
class OneflowDataLoader(DataLoader):
"""
+ 9
- 0
fastNLP/core/dataloaders/paddle_dataloader/fdl.py
View File
@@ -43,6 +43,15 @@ class _PaddleDataset(Dataset):
except Exception as e:
raise e
def __len__(self) -> int:
return len(self.dataset)
def __getstate__(self):
return self.__dict__
def __setstate__(self, state):
self.__dict__ = state
class PaddleDataLoader(DataLoader):
"""
Write
Preview
Loading…
Cancel
Save