Update mar.py
Browse files
mar.py
CHANGED
|
@@ -10,7 +10,7 @@ from torch.utils.checkpoint import checkpoint
|
|
| 10 |
|
| 11 |
from timm.models.vision_transformer import Block
|
| 12 |
|
| 13 |
-
from diffloss import DiffLoss
|
| 14 |
|
| 15 |
|
| 16 |
def mask_by_order(mask_len, order, bsz, seq_len):
|
|
|
|
| 10 |
|
| 11 |
from timm.models.vision_transformer import Block
|
| 12 |
|
| 13 |
+
from .diffloss import DiffLoss
|
| 14 |
|
| 15 |
|
| 16 |
def mask_by_order(mask_len, order, bsz, seq_len):
|