first commit
Some checks failed
Lint / quick-checks (push) Has been cancelled
Lint / flake8-py3 (push) Has been cancelled
Close inactive issues / close-issues (push) Successful in 33s

This commit is contained in:
2025-03-17 00:41:41 +08:00
commit ad1fee9c4b
411 changed files with 83607 additions and 0 deletions

View File

@@ -0,0 +1,5 @@
defaults:
- default
accelerator: cpu
devices: 1

View File

@@ -0,0 +1,9 @@
defaults:
- default
strategy: ddp
accelerator: gpu
devices: [0,1]
num_nodes: 1
sync_batchnorm: True

View File

@@ -0,0 +1,7 @@
defaults:
- default
# simulate DDP on CPU, useful for debugging
accelerator: cpu
devices: 2
strategy: ddp_spawn

View File

@@ -0,0 +1,20 @@
_target_: lightning.pytorch.trainer.Trainer
default_root_dir: ${paths.output_dir}
max_epochs: -1
accelerator: gpu
devices: [0]
# mixed precision for extra speed-up
precision: 16-mixed
# perform a validation loop every N training epochs
check_val_every_n_epoch: 1
# set True to to ensure deterministic results
# makes training slower but gives more reproducibility than just setting seeds
deterministic: False
gradient_clip_val: 5.0

View File

@@ -0,0 +1,5 @@
defaults:
- default
accelerator: gpu
devices: 1

View File

@@ -0,0 +1,5 @@
defaults:
- default
accelerator: mps
devices: 1