Mirrors_Framework/deepmind-research
1
0
Fork 0
mirror of https://github.com/google-deepmind/deepmind-research.git synced 2026-05-28 10:45:46 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add checkpoints from the ablation study.

Browse Source 
PiperOrigin-RevId: 328023346
This commit is contained in:
Florent Altché
2020-08-23 14:26:26 +01:00
committed by Diego de Las Casas
parent 22c3daff19
commit 8457046b2c
33 changed files with 397 additions and 363 deletions
Download Patch File Download Diff File Expand all files Collapse all files
geomancer/train.py
+2 -2
View File
@@ -46,7 +46,7 @@ def make_so_tangent(q):
for j in range(i+1, n):
a[i, j] = 1
a[j, i] = -1
dq[Ellipsis, ii] = a @ q # tangent vectors are skew-symmetric matrix times Q
dq[..., ii] = a @ q # tangent vectors are skew-symmetric matrix times Q
a[i, j] = 0
a[j, i] = 0
ii += 1
@@ -106,7 +106,7 @@ def make_product_manifold(specification, npts):
spec_array[1, i] = dim
latent_dim += dim
dat = np.random.randn(npts, dim+1)
dat /= np.tile(np.sqrt(np.sum(dat**2, axis=1)[Ellipsis, None]),
dat /= np.tile(np.sqrt(np.sum(dat**2, axis=1)[..., None]),
[1, dim+1])
elif so_spec is not None:
dim = int(so_spec.group(1))