fixes

.gitattributes

@@ -35,3 +35,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 utils/res10_300x300_ssd_iter_140000_fp16.caffemodel filter=lfs diff=lfs merge=lfs -text
 core/vision_encoder/bpe_simple_vocab_16e6.txt.gz filter=lfs diff=lfs merge=lfs -text
+utils/res10_300x300_ssd_iter_140000_fp16.caffemodel filter=lfs diff=lfs merge=lfs -text

src/model.py

@@ -18,7 +18,7 @@ DROPOUT_P = 0.5
 
 
 class MTLModel(nn.Module):
-    def __init__(self, backbone, tasks: List[Task], device
+    def __init__(self, backbone, tasks: List[Task], device,
                 rank: int = 64, 
                 use_lora: bool = True, 
                 truncate_idx: int = 22, 

utils/res10_300x300_ssd_iter_140000_fp16.caffemodel

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:510ffd2471bd81e3fcc88a5beb4eae4fb445ccf8333ebc54e7302b83f4158a76
+size 5351047