屏蔽不必要告警
屏蔽不必要告警
This commit is contained in:
@@ -1,5 +1,6 @@
|
||||
import warnings
|
||||
warnings.filterwarnings("ignore")
|
||||
import utils, os
|
||||
|
||||
hps = utils.get_hparams(stage=2)
|
||||
os.environ["CUDA_VISIBLE_DEVICES"] = hps.train.gpu_numbers.replace("-", ",")
|
||||
import torch
|
||||
|
||||
Reference in New Issue
Block a user