Skip to content

Commit c8ac10f

Browse files
committed
add higgs files
1 parent 79211d7 commit c8ac10f

File tree

3 files changed

+1
-1039
lines changed

3 files changed

+1
-1039
lines changed

higgs_runner_clock.py renamed to higgs_runner.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -40,13 +40,10 @@
4040
from fuse.eval.metrics.classification.metrics_classification_common import MetricAccuracy, MetricAUCROC, MetricROCCurve
4141
from fuse.data.utils.collates import CollateDefault
4242
from fuse.data.utils.samplers import BatchSamplerDefault
43-
from fuse.dl.models.backbones.backbone_mlp import BackboneMultilayerPerceptron
4443
from fuse.dl.models.heads.head_global_pooling_classifier import HeadGlobalPoolingClassifier
4544
from fuse.dl.models.backbones.backbone_resnet import BackboneResnet
4645
from fuse.dl.models.backbones.backbone_inception_resnet_v2 import BackboneInceptionResnetV2
4746

48-
from fuse.dl.models.heads.common import ClassifierMLP
49-
5047
from fuse.dl.models import ModelMultiHead
5148
from fuse.dl.lightning.pl_module import LightningModuleDefault
5249
from fuse.dl.lightning.pl_funcs import convert_predictions_to_dataframe
@@ -55,8 +52,7 @@
5552
from fuse.eval.evaluator import EvaluatorDefault
5653

5754
from higgs import HIGGS
58-
from utils.autoencoder import Encoder, Decoder, OurEncodingLoss
59-
import torchvision.models as models
55+
6056

6157
###########################################################################################################
6258
# Fuse
@@ -70,8 +66,6 @@
7066
experiment = "TAC_CLOCK" # Choose from supported experiments
7167

7268
supported_experiments = [
73-
"TAC_BRICK",
74-
"TAC_CAMERA",
7569
"TAC_CLOCK",
7670
]
7771
assert experiment in supported_experiments, f"runner doesn't support experiment ({experiment})."

0 commit comments

Comments
 (0)