alihan/textattack-nlp-transformer
1
0
Fork 0
mirror of https://github.com/QData/TextAttack.git synced 2021-10-13 00:05:06 +03:00
Code Issues Projects Releases Wiki Activity

add missing logger

Browse Source
This commit is contained in:
Jin Yong Yoo
2020-05-23 04:55:29 -04:00
parent a5dfc23894
commit 0191c4ac5c
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
textattack/shared/scripts/run_attack_parallel.py
View File

@@ -10,6 +10,8 @@ import tqdm
from .run_attack_args_helper import *
logger = textattack.shared.utils.get_logger()
def set_env_variables(gpu_id):
# Set sharing strategy to file_system to avoid file descriptor leaks
torch.multiprocessing.set_sharing_strategy('file_system')