VITRA / vitra /training /__init__.py
arnoldland's picture
Initial commit
aae3ba1
from .base_strategy import TrainingStrategy
from .fsdp import VLAFSDPStrategy
from .metrics import Metrics, VLAMetrics