Add missing task fields to the models

torchbenchmark/models/mobilenet_v2/__init__.py

@@ -14,6 +14,7 @@
 #######################################################
 class Model(BenchmarkModel):
     task = COMPUTER_VISION.CLASSIFICATION
+
     def __init__(self, device=None, jit=False):
         super().__init__()
         self.device = device

torchbenchmark/models/mobilenet_v2_quantized_qat/__init__.py

@@ -3,10 +3,13 @@
 import torch.optim as optim
 import torchvision.models as models
 from torch.quantization import quantize_fx
+from torchbenchmark.tasks import COMPUTER_VISION
 from ...util.model import BenchmarkModel
 
 
 class Model(BenchmarkModel):
+    task = COMPUTER_VISION.CLASSIFICATION
+
     def __init__(self, device=None, jit=False):
         super().__init__()
         self.device = device

torchbenchmark/models/pyhpc_equation_of_state/__init__.py

@@ -1,5 +1,6 @@
 import torch
 from . import eos_pytorch
+from torchbenchmark.tasks import OTHER
 from ...util.model import BenchmarkModel
 
 def _generate_inputs(size):
@@ -26,6 +27,8 @@ def forward(self, s, t, p):
         return eos_pytorch.gsw_dHdT(s, t, p)
 
 class Model(BenchmarkModel):
+    task = OTHER.OTHER_TASKS
+
     def __init__(self, device=None, jit=False):
         super().__init__()
         self.device = device

torchbenchmark/models/pyhpc_isoneutral_mixing/__init__.py

@@ -1,5 +1,6 @@
 import torch
 from . import isoneutral_pytorch
+from torchbenchmark.tasks import OTHER
 from ...util.model import BenchmarkModel
 
 
@@ -122,6 +123,8 @@ def forward(
 
 
 class Model(BenchmarkModel):
+    task = OTHER.OTHER_TASKS
+
     def __init__(self, device=None, jit=False):
         super().__init__()
         self.device = device