Skip to content

Commit 91dd9a6

Browse files
Update pmoe.py
1 parent 095a331 commit 91dd9a6

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

pmoe.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,9 @@
11
'''
22
Probabilistic Mixture-of-Experts
33
paper: https://arxiv.org/abs/2104.09122
4+
5+
Core features:
6+
It replaces the diagonal Gaussian distribution with (differentiable) Gaussian mixture model for policy function approximation, which is more expressive.
47
'''
58

69
import argparse

0 commit comments

Comments
 (0)