8 lines
182 B
Python
8 lines
182 B
Python
from src.model.utransformer import UTransformer
|
|
|
|
test_model = UTransformer.from_pretrained_backbone(
|
|
"facebook/dinov3-vits16-pretrain-lvd1689m"
|
|
).to("cuda:3")
|
|
|
|
print(test_model)
|