IF YOU WOULD LIKE TO GET AN ACCOUNT, please write an email to s dot adaszewski at gmail dot com. User accounts are meant only to report issues and/or generate pull requests. This is a purpose-specific Git hosting for ADARED projects. Thank you for your understanding!
Ви не можете вибрати більше 25 тем Теми мають розпочинатися з літери або цифри, можуть містити дефіси (-) і не повинні перевищувати 35 символів.

14 рядки
416B

  1. import torch
  2. import numpy as np
  3. def init_glorot(input_dim, output_dim):
  4. """Create a weight variable with Glorot & Bengio (AISTATS 2010)
  5. initialization.
  6. """
  7. init_range = np.sqrt(6.0 / (input_dim + output_dim))
  8. initial = -init_range + 2 * init_range * \
  9. torch.rand(( input_dim, output_dim ), dtype=torch.float32)
  10. initial = initial.requires_grad_(True)
  11. return initial