fixed problems with variational dropout

2020-05-25 14:30:45 -04:00
parent dd91321527
commit 3fac6470f4
5 changed files with 22 additions and 8 deletions
--- a/core/raft.py
+++ b/core/raft.py
@@ -26,7 +26,7 @@ class RAFT(nn.Module):
            args.corr_levels = 4
            args.corr_radius = 4

-        if 'dropout' not in args._get_kwargs():
+        if not hasattr(args, 'dropout'):
            args.dropout = 0

        # feature network, context network, and update block