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
|