Cannot repeat results when using cox task

[dengyq365]

model = stg.STG(task_type='cox',input_dim=train_data_K['X'].shape[1], output_dim=1, hidden_dims=[60, 20, 2],activation='selu', optimizer='Adam', learning_rate=1e-3, batch_size=train_data_K['X'].shape[0], feature_selection=feature_selection,  sigma=0.5, lam=0.001, random_state=123, device=device)
model.fit(train_data_K['X'], {'E': train_data_K['E'],'T': train_data_K['T']},
          valid_X=val_data_K['X'], valid_y= {'E': val_data_K['E'], 'T': val_data_K['T']}, 
          nr_epochs=460, print_interval=100)
## get gates
np.where(model.get_gates(mode='prob'))

could not get the same result when I repeat only above code.