diff --git a/main/_images/RReLU.png b/main/_images/RReLU.png index ead5d91c89..6c8299b605 100644 Binary files a/main/_images/RReLU.png and b/main/_images/RReLU.png differ diff --git a/main/_modules/index.html b/main/_modules/index.html index 647749d227..956d392af4 100644 --- a/main/_modules/index.html +++ b/main/_modules/index.html @@ -280,7 +280,7 @@
mod = getattr(mod, item)
- if type(mod) is not torch.nn.Module:
+ # Use isinstance instead of type here to also handle subclass of nn.Module
+ if not isinstance(mod, torch.nn.Module):
raise AttributeError("`" + item + "` is not an nn.Module")
+
setattr(mod, name, module)