From 39601603ef9a21abf5fbc571726eeca623307b16 Mon Sep 17 00:00:00 2001
From: jh <jmhummel@gmail.com>
Date: Fri, 12 Mar 2021 18:32:49 -0800
Subject: [PATCH] no message

---
 core/leras/layers/MsSsim.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/core/leras/layers/MsSsim.py b/core/leras/layers/MsSsim.py
index 84028bb..b631491 100644
--- a/core/leras/layers/MsSsim.py
+++ b/core/leras/layers/MsSsim.py
@@ -22,7 +22,7 @@ class MsSsim(nn.LayerBase):
 
 
         def assign_device(op):
-            if op.type != 'Assert':
+            if op.type != 'Assert' or op.type != 'ListDiff':
                 return '/gpu:0'
             else:
                 return '/cpu:0'