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