Update src/model/encoder/anysplat.py
Browse files
src/model/encoder/anysplat.py
CHANGED
|
@@ -130,8 +130,7 @@ class EncoderAnySplat(Encoder[EncoderAnySplatCfg]):
|
|
| 130 |
|
| 131 |
def __init__(self, cfg: EncoderAnySplatCfg) -> None:
|
| 132 |
super().__init__(cfg)
|
| 133 |
-
model_full = VGGT
|
| 134 |
-
# model_full = VGGT()
|
| 135 |
self.aggregator = model_full.aggregator.to(torch.bfloat16)
|
| 136 |
self.freeze_backbone = cfg.freeze_backbone
|
| 137 |
self.distill = cfg.distill
|
|
|
|
| 130 |
|
| 131 |
def __init__(self, cfg: EncoderAnySplatCfg) -> None:
|
| 132 |
super().__init__(cfg)
|
| 133 |
+
model_full = VGGT()
|
|
|
|
| 134 |
self.aggregator = model_full.aggregator.to(torch.bfloat16)
|
| 135 |
self.freeze_backbone = cfg.freeze_backbone
|
| 136 |
self.distill = cfg.distill
|