I am trying to import PyroModule using: from pyro.nn import PyroModule
but, I am getting this error:
ImportError: cannot import name 'PyroModule' from 'pyro.nn' (/home/karima/anaconda3/envs/my_env/lib/python3.7/site-packages/pyro/nn/init.py)
Here is the whole piece of code:
import os
from functools import partial
import torch
import numpy as np
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
import pyro
import pyro.distributions as dist
from torch import nn
import seaborn as sns
from pyro.nn import PyroModule
I think the issue is due to wrong installation of pyro module,
instead of this
can you try this
Let me know how this goes