Ssd.pytorch: NameError: name 'viz' is not defined

[sxyxf66]

[JigyasaK]

I solved this by moving the following statements to the top, with other import statements.

import torch.backends.cudnn as cudnn
import torch.nn.init as init
import torch.utils.data as data
import numpy as np
import argparse
import visdom

viz = visdom.Visdom()

[shirui-japina]

I solved this by moving the following statements to the top, with other import statements.

import torch.backends.cudnn as cudnn
import torch.nn.init as init
import torch.utils.data as data
import numpy as np
import argparse
import visdom

viz = visdom.Visdom()

YES!! Worked like below:

import ...

parser.add_argument()
...

args = parser.parse_args()

if args.visdom:
    import visdom
    viz = visdom.Visdom()