↓ 16 callersMethod__init__(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
model/vision_transformer.py:49
↓ 5 callersFunctiontest_kmeans(model, test_loader,
epoch, save_name, device,
args, logger_class=None, output
project_utils/k_means_utils.py:215
↓ 2 callersMethod__init__(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
model/attribute_classifier.py:65
↓ 1 callersMethod_gather_outputs(self, field, log_fields, stat_parent, stat, require_dict=False)
project_utils/visualization_utils.py:1132