tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Reduce bert_gen numprocess to 2 (often freezes in colab)

Browse Source
This commit is contained in:
litagin02
2024-01-06 15:07:44 +09:00
parent 100949d970
commit 09c0673ded
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
default_config.yml
View File

@@ -22,7 +22,7 @@ preprocess_text:
bert_gen: bert_gen:
config_path: "config.json" config_path: "config.json"
num_processes: 4 num_processes: 2
device: "cuda" device: "cuda"
use_multi_device: false use_multi_device: false