天天看點

torch.manual_seed( )

在神經網絡中,參數預設是進行随機初始化的。如果不設定的話每次訓練時的初始化都是随機的,導緻結果不确定。如果設定初始化,則每次初始化都是固定的。

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           

繼續閱讀