This commit is contained in:
neulus
2025-09-29 16:00:05 +09:00
parent 2761171fe3
commit d3793890a7
6 changed files with 189 additions and 24 deletions

View File

@@ -1,6 +1,7 @@
def main():
print("Hello from cloud-removal!")
from src.model.utransformer import UTransformer
test_model = UTransformer.from_pretrained_backbone(
"facebook/dinov3-vits16-pretrain-lvd1689m"
).to("cuda:3")
if __name__ == "__main__":
main()
print(test_model)