joel1895

This will solve the issue, after model.initialize(), add the below code .Here,the model is wrapped around the DataParallel and the 'module' attribute is coming from here. `model = torch.nn.DataParallel(model, device_ids=[0])...

I have used the hair_seg.pth given in the repo in the run_eval.py, it's giving me key errors. I used checkpoint model.h5 and still I didn't get results. Is the pre-trained...

# Create train data: python generate_tfrecord.py --csv_input=data/train_labels.csv --output_path=data/train.record --image_dir=images/train/ # Create test data: python generate_tfrecord.py --csv_input=data/test_labels.csv --output_path=data/test.record --image_dir=images/test/ try this, worked for me

I am also facing the error, any solution for this?