Merge branch 'master' of https://github.com/autonlab/DeepSurvivalMach…

…ines
autonlab · Dec 24, 2021 · 77ba704 · 77ba704
2 parents 9b088d0 + 9454fae
commit 77ba704
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -192,5 +192,5 @@ FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TOR
 OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 DEALINGS IN THE SOFTWARE.
 
-<img style="float: left; padding-top:25px" width ="150px" src="https://www.cmu.edu/brand/downloads/assets/images/wordmarks-600x600-min.jpg">
-<img style="float: right; padding-top:25px" width = "150px" src="https://www.autonlab.org/user/themes/auton/images/AutonLogo.png"> 
+<img align="right" height ="120px" src="https://www.cs.cmu.edu/~chiragn/cmu_logo.jpeg">
+<img align="right" height ="110px" src="https://www.cs.cmu.edu/~chiragn/auton_logo.png"> 
diff --git a/dsm/contrib/dcm_torch.py b/dsm/contrib/dcm_torch.py
@@ -58,4 +58,4 @@ def forward(self, x):
       log_hazard_ratios = torch.clamp(self.expert(x), min=-gamma, max=gamma)
     log_gate_prob = torch.nn.LogSoftmax(dim=1)(self.gate(x))
 
-    return log_gate_prob, log_hazard_ratios
+    return log_gate_prob, log_hazard_ratios