Update train_nsf_sim_cache_sid_load_pretrain.py

2023-05-24 12:27:15 +00:00 · 2023-05-24 12:27:15 +00:00 · e0813eb282
parent 8efb101401
commit e0813eb282
1 changed files with 585 additions and 584 deletions
--- a/train_nsf_sim_cache_sid_load_pretrain.py
+++ b/train_nsf_sim_cache_sid_load_pretrain.py
@ -581,4 +581,5 @@ def train_and_evaluate(


 if __name__ == "__main__":
+    torch.multiprocessing.set_start_method('spawn')
    main()