[BUG]

[wangqiankun0201]

C:\Users\ASUS\anaconda3\envs\lunwen\python.exe E:\my_paper_code\models\FL-Transformer.py
C:\Users\ASUS\anaconda3\envs\lunwen\Lib\site-packages\torch\nn\modules\transformer.py:382: UserWarning: enable_nested_tensor is True, but self.use_nested_tensor is False because encoder_layer.self_attn.batch_first was not True(use batch_first for better inference performance)
warnings.warn(
Traceback (most recent call last):
File "E:\my_paper_code\models\FL-Transformer.py", line 48, in
regressor.fit(X_train, y_train)
^^^^^^^^^^^^^
File "C:\Users\ASUS\anaconda3\envs\lunwen\Lib\site-packages\torch\nn\modules\module.py", line 1940, in getattr
raise AttributeError(
AttributeError: 'FTTransformer' object has no attribute 'fit'