File tree Expand file tree Collapse file tree 4 files changed +3
-1
lines changed Expand file tree Collapse file tree 4 files changed +3
-1
lines changed Original file line number Diff line number Diff line change @@ -8,6 +8,7 @@ training:
88 progress_bar_refresh_rate : 50
99 val_check_interval : 1000
1010 num_sanity_val_steps : 0
11+ strategy : ddp
1112 lightning_checkpoint :
1213 dirpath : " ."
1314 filename : flava-{epoch:02d}-{step}
Original file line number Diff line number Diff line change @@ -8,6 +8,7 @@ training:
88 progress_bar_refresh_rate : 50
99 val_check_interval : 100
1010 num_sanity_val_steps : 0
11+ strategy : ddp
1112 lightning_checkpoint :
1213 dirpath : " ."
1314 filename : flava-{epoch:02d}-{step}
Original file line number Diff line number Diff line change @@ -7,6 +7,7 @@ training:
77 progress_bar_refresh_rate : 50
88 val_check_interval : 10000
99 num_sanity_val_steps : 0
10+ strategy : ddp
1011 lightning_checkpoint :
1112 dirpath : " ."
1213 filename : flava-{epoch:02d}-{step}
Original file line number Diff line number Diff line change @@ -68,7 +68,6 @@ def main():
6868 trainer = Trainer (
6969 ** OmegaConf .to_container (config .training .lightning ),
7070 callbacks = callbacks ,
71- strategy = "ddp" ,
7271 )
7372 ckpt_path = config .training .lightning_load_from_checkpoint
7473
You can’t perform that action at this time.
0 commit comments