From aa3d1f258ad05dd1aa3a05f5352987d4598280fc Mon Sep 17 00:00:00 2001
From: Kashif Rasul <kashif.rasul@gmail.com>
Date: Fri, 1 Apr 2022 19:48:28 +0200
Subject: [PATCH] fix name

---
 hopfield/module.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/hopfield/module.py b/hopfield/module.py
index d647d9b..a3c67a9 100644
--- a/hopfield/module.py
+++ b/hopfield/module.py
@@ -13,7 +13,7 @@ from hflayers import Hopfield
 from hflayers.transformer import HopfieldDecoderLayer, HopfieldEncoderLayer
 
 
-class HopfieldNetwork(nn.Module):
+class HopfieldModel(nn.Module):
     @validated()
     def __init__(
         self,