mirror of
https://github.com/wassname/pytorch-transformer-ts.git
synced 2026-06-27 18:06:14 +08:00
fix name
This commit is contained in:
+1
-1
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user