Olamide Isreal
a3ffec6a07
Some checks failed
Code Quality Main / code-quality (push) Has been cancelled
Release Drafter / update_release_draft (push) Has been cancelled
Tests / run_tests_ubuntu (ubuntu-latest, 3.10) (push) Has been cancelled
Tests / run_tests_ubuntu (ubuntu-latest, 3.8) (push) Has been cancelled
Tests / run_tests_ubuntu (ubuntu-latest, 3.9) (push) Has been cancelled
Tests / run_tests_macos (macos-latest, 3.10) (push) Has been cancelled
Tests / run_tests_macos (macos-latest, 3.8) (push) Has been cancelled
Tests / run_tests_macos (macos-latest, 3.9) (push) Has been cancelled
Tests / run_tests_windows (windows-latest, 3.10) (push) Has been cancelled
Tests / run_tests_windows (windows-latest, 3.8) (push) Has been cancelled
Tests / run_tests_windows (windows-latest, 3.9) (push) Has been cancelled
Tests / code-coverage (push) Has been cancelled
13 lines
416 B
YAML
13 lines
416 B
YAML
_target_: lightning.pytorch.strategies.FSDPStrategy
|
|
sharding_strategy: ${resolve_variable:torch.distributed.fsdp.ShardingStrategy.FULL_SHARD}
|
|
cpu_offload: null
|
|
activation_checkpointing: null
|
|
mixed_precision:
|
|
_target_: torch.distributed.fsdp.MixedPrecision
|
|
param_dtype: null
|
|
reduce_dtype: null
|
|
buffer_dtype: null
|
|
keep_low_precision_grads: false
|
|
cast_forward_inputs: false
|
|
cast_root_forward_inputs: true
|