File size: 359 Bytes
0613ebe | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | # Copyright 2024-2025 The Robbyant Team Authors. All rights reserved.
import torch
from easydict import EasyDict
va_shared_cfg = EasyDict()
va_shared_cfg.host = '0.0.0.0'
va_shared_cfg.port = 29536
va_shared_cfg.param_dtype = torch.bfloat16
va_shared_cfg.save_root = './train_out'
va_shared_cfg.patch_size = (1, 2, 2)
va_shared_cfg.enable_offload = False |