在神經網絡中,參數預設是進行随機初始化的。如果不設定的話每次訓練時的初始化都是随機的,導緻結果不确定。如果設定初始化,則每次初始化都是固定的。
if args.seed is not None:
random.seed(args.seed) #
torch.manual_seed(args.seed) #為CPU設定種子用于生成随機數,以使得結果是确定的
torch.cuda.manual_seed(args.seed) #為目前GPU設定随機種子;
cudnn.deterministic = True