thienphuc12339 commited on
Commit
d1bce21
·
verified ·
1 Parent(s): 45b6bc8

Update tools/features.py

Browse files
Files changed (1) hide show
  1. tools/features.py +1 -1
tools/features.py CHANGED
@@ -2,7 +2,7 @@
2
 
3
  import torch
4
  from configs import DataConfig
5
- from features import BaseDataset, VSL98Dataset, VSL400Dataset
6
 
7
 
8
  def load_dataset(data_config: DataConfig) -> BaseDataset:
 
2
 
3
  import torch
4
  from configs import DataConfig
5
+ from tools.features import BaseDataset, VSL98Dataset, VSL400Dataset
6
 
7
 
8
  def load_dataset(data_config: DataConfig) -> BaseDataset: