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!
sadaszewski1
/
decagon-pytorch
forked from sadaszewski/decagon-pytorch
1
0
Fork 0
Code Issues 0 Pull Requests 0 Releases 0 Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
73 Commits
1 Branch
866KB
Tree: 6d0039b447
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '6d0039b447'
${ noResults }
decagon-pytorch/tests/icosagon/test_weights.py

14 lines
363B
Raw Blame History 

  1. from icosagon.weights import init_glorot

  2. import torch

  3. import numpy as np

  4. 

  5. 

  6. def test_init_glorot_01():

  7.     torch.random.manual_seed(0)

  8.     res = init_glorot(10, 20)

  9.     torch.random.manual_seed(0)

  10.     rnd = torch.rand((10, 20))

  11.     init_range = np.sqrt(6.0 / 30)

  12.     expected = -init_range + 2 * init_range * rnd

  13.     assert torch.all(res == expected)