10 lines
316 B
Plaintext
10 lines
316 B
Plaintext
# RTX 4090 profile, shared encoder (DINOv3).
|
|
HardwareConfig.device = 'cuda'
|
|
HardwareConfig.batch_size = 8
|
|
HardwareConfig.grad_accum_steps = 8
|
|
HardwareConfig.num_workers = 4
|
|
HardwareConfig.use_amp = True
|
|
HardwareConfig.gradient_checkpointing = True
|
|
HardwareConfig.reserve_gb = 2.0
|
|
HardwareConfig.total_vram_gb = 24.0
|