fixed import mlp

desResLab · Nov 7, 2023 · df0dcdc · df0dcdc
1 parent 4eb21fd
commit df0dcdc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/linfa/nofas.py b/linfa/nofas.py
@@ -2,7 +2,7 @@
 import torch.nn as nn
 import torch.nn.functional as F
 import numpy as np
-from mlp import FNN
+from linfa.mlp import FNN
 from os import path
 
 torch.set_default_tensor_type(torch.DoubleTensor)