This commit is contained in:
Kashif Rasul
2022-04-01 19:48:28 +02:00
parent 2267212627
commit aa3d1f258a
+1 -1
View File
@@ -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,