Skip to content

Commit a72b9bd

Browse files
committed
Fix detectron2
1 parent 1112339 commit a72b9bd

File tree

1 file changed

+0
-14
lines changed

1 file changed

+0
-14
lines changed

torchbenchmark/models/detectron2_maskrcnn/__init__.py

Lines changed: 0 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -31,22 +31,8 @@ class Model(BenchmarkModel):
3131
DEFAULT_TRAIN_BSIZE = 1
3232
DEFAULT_EVAL_BSIZE = 2
3333

34-
<<<<<<< HEAD
3534
def __init__(self, test, device, jit=False, batch_size=None, extra_args=[]):
3635
super().__init__(test=test, device=device, jit=jit, batch_size=batch_size, extra_args=extra_args)
37-
=======
38-
def __init__(self, device=None, jit=False, train_bs=1, eval_bs=2):
39-
super().__init__()
40-
self.device = device
41-
self.jit = jit
42-
43-
model_cfg = model_zoo.get_config("common/models/mask_rcnn_fpn.py").model
44-
self.model = instantiate(model_cfg).to(self.device)
45-
self.train_bs = train_bs
46-
self.eval_bs = eval_bs
47-
48-
data_cfg = model_zoo.get_config("common/data/coco.py").dataloader
49-
>>>>>>> 03f3720 (Downscale to single GPU with batch size == 16.)
5036

5137
model_cfg = model_zoo.get_config("common/models/mask_rcnn_fpn.py").model
5238
data_cfg = model_zoo.get_config("common/data/coco.py").dataloader

0 commit comments

Comments
 (0)