ourownstory / ar-net Goto Github PK

Cell In[1], [line 1](vscode-notebook-cell:?execution_count=1&line=1)
----> [1](vscode-notebook-cell:?execution_count=1&line=1) import arnet

File .\arnet\__init__.py:24
     [21](./arnet/__init__.py:21) from fastai.basics import *
     [22](./arnet/__init__.py:22) from fastai.tabular.all import *
---> [24](./arnet/__init__.py:24) from .ar_net import ARNet
     [26](.arnet/__init__.py:26) # from .ar_net_legacy import init_ar_learner
     [27](./arnet/__init__.py:27) from .utils_data import load_from_file, tabularize_univariate, estimate_noise, split_by_p_valid

File .\arnet\ar_net.py:13
     [11](./arnet/ar_net.py:11) from fastai.data.transforms import Normalize
     [12](./arnet/ar_net.py:12) from fastai.learner import load_learner
---> [13](./arnet/ar_net.py:13) from fastai.distributed import ParallelTrainer
     [15](./arnet/ar_net.py:15) from arnet import utils, utils_data, plotting
     [16](./arnet/ar_net.py:16) from arnet.fastai_mods import SparsifyAR, huber, sTPE, get_loss_func

File .\fastai\distributed.py:139
    [136](./fastai/distributed.py:136) _hidden_params = ["mixed_precision", "fp16", "log_with", "logging_dir", "step_scheduler_with_optimizer"]
    [138](./fastai/distributed.py:138) # %% ../nbs/20a_distributed.ipynb 30
--> [139](./fastai/distributed.py:139) class DistributedTrainer(Callback):
    [140](./fastai/distributed.py:140)     "Wrap `model` in `DistributedDataParallel` and `dls` in `DistributedDL`"
    [141](./fastai/distributed.py:141)     order = 11
...
    [146](./fastai/distributed.py:146) ):
    [147](./fastai/distributed.py:147)     store_attr()
    [148](./fastai/distributed.py:148)     self.accelerator = Accelerator(**kwargs)

NameError: name 'Accelerator' is not defined