Giters

TUI-NICR / ESANet

ESANet: Efficient RGB-D Semantic Segmentation for Indoor Scene Analysis

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

logs.csv doesn't exist

sangwonme opened this issue · comments

commented

I'm having trouble with training ESA.

Pretrained weights link(https://drive.google.com/uc?id=1neUb6SJ87dIY1VvrSGxurVBQlH8Pd_Bi) provided here only has r34_NBt1D.pth file not the logs.csv file. So when I run the train.py, it shows like below.

FileNotFoundError: [Errno 2] No such file or directory: './trained_models/imagenet/logs.csv'
Where can I get logs.csv file?
Is there any possible way except for running imagenet_pretraining.py by myself?

commented

What do you need the logs for? Can you please provide the full call and error log?

commented

Did you forgot to save the weights in the proper directory? By default the weights should be saved here: ./trained_models/imagenet/r34_NBt1D.pth, as you can see in the code:

ESANet/src/models/resnet.py

Line 473 in a271ac5

ckpt_path = os.path.join(pretrained_dir, f'{resnet_name}_NBt1D.pth')

Only if these weights are not found, the function tries to load the logs.

commented

Oh thanks!
It was problem caused by wrong directory.
Now I can run train code!

Thanks a lot :)