Traceback (most recent call last):


 "C:\Users\adnan\AppData\Local\Programs\Python\Python38-32\lib\site-packages\tornado\ioloop.py", line 907, in _run

return self.callback()


 "C:\Users\adnan\AppData\Local\Programs\Python\Python38-32\lib\site-packages\distributed\nanny.py", line 414, in memory_monitor

process = self.process.process
AttributeError: 'NoneType' object has no attribute 'process'

Traceback (most recent call last):


 "test.py", line 62, in <module>

trainer.fit(model, mnist_train)


 "/home/ubuntu/anaconda3/lib/python3.7/site-packages/mlflow/pytorch/_pytorch_autolog.py", line 216, in fit

return _run_and_log_function(self, original, args, kwargs)


 "/home/ubuntu/anaconda3/lib/python3.7/site-packages/mlflow/pytorch/_pytorch_autolog.py", line 205, in _run_and_log_function

result = original(self, *args, **kwargs)


 "/home/ubuntu/anaconda3/lib/python3.7/site-packages/pytorch_lightning/trainer/trainer.py", line 439, in fit

results = self.accelerator_backend.train()


 "/home/ubuntu/anaconda3/lib/python3.7/site-packages/pytorch_lightning/accelerators/cpu_accelerator.py", line 48, in train

results = self.train_or_test()


 "/home/ubuntu/anaconda3/lib/python3.7/site-packages/pytorch_lightning/accelerators/accelerator.py", line 66, in train_or_test

results = self.trainer.train()


 "/home/ubuntu/anaconda3/lib/python3.7/site-packages/pytorch_lightning/trainer/trainer.py", line 487, in train

self.train_loop.on_train_end()


 "/home/ubuntu/anaconda3/lib/python3.7/site-packages/pytorch_lightning/trainer/training_loop.py", line 182, in on_train_end

self.trainer.call_hook('on_train_end')


 "/home/ubuntu/anaconda3/lib/python3.7/site-packages/pytorch_lightning/trainer/trainer.py", line 822, in call_hook

trainer_hook(*args, **kwargs)


 "/home/ubuntu/anaconda3/lib/python3.7/site-packages/pytorch_lightning/trainer/callback_hook.py", line 117, in on_train_end

callback.on_train_end(self, self.get_model())


 "/home/ubuntu/anaconda3/lib/python3.7/site-packages/mlflow/pytorch/_pytorch_autolog.py", line 124, in on_train_end

mlflow.pytorch.log_model(pytorch_model=trainer.model, artifact_path="model")


 "/home/ubuntu/anaconda3/lib/python3.7/site-packages/mlflow/pytorch/__init__.py", line 239, in log_model

**kwargs,


 "/home/ubuntu/anaconda3/lib/python3.7/site-packages/mlflow/models/model.py", line 172, in log

flavor.save_model(path=local_path, mlflow_model=mlflow_model, **kwargs)


 "/home/ubuntu/anaconda3/lib/python3.7/site-packages/mlflow/pytorch/__init__.py", line 405, in save_model

torch.save(pytorch_model, model_path, pickle_module=pickle_module, **kwargs)


 "/home/ubuntu/anaconda3/lib/python3.7/site-packages/torch/serialization.py", line 372, in save

_save(obj, opened_zipfile, pickle_module, pickle_protocol)


 "/home/ubuntu/anaconda3/lib/python3.7/site-packages/torch/serialization.py", line 476, in _save

pickler.dump(obj)


 "/home/ubuntu/anaconda3/lib/python3.7/site-packages/cloudpickle/cloudpickle.py", line 357, in dump

return Pickler.dump(self, obj)


 "/home/ubuntu/anaconda3/lib/python3.7/pickle.py", line 437, in dump

self.save(obj)


 "/home/ubuntu/anaconda3/lib/python3.7/pickle.py", line 549, in save

self.save_reduce(obj=obj, *rv)


 "/home/ubuntu/anaconda3/lib/python3.7/pickle.py", line 662, in save_reduce

save(state)


 "/home/ubuntu/anaconda3/lib/python3.7/pickle.py", line 504, in save

f(self, obj) # Call unbound method with explicit self


 "/home/ubuntu/anaconda3/lib/python3.7/pickle.py", line 859, in save_dict

self._batch_setitems(obj.items())


 "/home/ubuntu/anaconda3/lib/python3.7/pickle.py", line 885, in _batch_setitems

save(v)


 "/home/ubuntu/anaconda3/lib/python3.7/pickle.py", line 535, in save

self.save_global(obj, rv)


 "/home/ubuntu/anaconda3/lib/python3.7/site-packages/cloudpickle/cloudpickle.py", line 837, in save_global

if obj.__module__ == "__main__":
AttributeError: 'torch.dtype' object has no attribute '__module__'
