8 lines
254 B
Plaintext
8 lines
254 B
Plaintext
# RTX 4090 profile.
|
|
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 |