Browse Source

[FIX] fix wrong delete

tags/v0.3.2
liuht 2 years ago
parent
commit
2ff7e15248
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      learnware/market/heterogeneous/organizer/hetero_map/__init__.py

+ 1
- 1
learnware/market/heterogeneous/organizer/hetero_map/__init__.py View File

@@ -9,7 +9,7 @@ from torch import Tensor, nn
from .....utils import allocate_cuda_idx, choose_device
from .....specification import HeteroMapTableSpecification, RKMETableSpecification
from .feature_extractor import CLSToken, FeatureProcessor, FeatureTokenizer
from .trainer import TransTabCollatorForCL
from .trainer import TransTabCollatorForCL, Trainer


class HeteroMap(nn.Module):


Loading…
Cancel
Save