search

yu4u / age-estimation-pytorch

PyTorch-based CNN implementation for estimating age from face images
MIT License
299 stars 86 forks source link

where is the output graph? #12

Open Bosman1 opened 4 years ago

Bosman1 commented 4 years ago

Hi, i run this command "python train.py --data_dir [PATH/TO/appa-real-release] --tensorboard tf_log MODEL.ARCH se_resnet50 TRAIN.OPT sgd TRAIN.LR 0.1" without error but i dont know where the output graph was saved. Pls help out.

Thanks

yu4u commented 4 years ago

'checkpoint' directory should be there

Bosman1 commented 4 years ago

okay. but i only have the weight for each epoch in the checkpoint directory.

how will i generate the graph with the those weight files?

Thanks

yu4u commented 4 years ago

see test.py

https://github.com/yu4u/age-estimation-pytorch/blob/master/test.py