sunxm2357/VideoIQ

Different definitions of policy network for different commands

Opened this issue · 0 comments

Thanks for sharing this great resources. I have a question of policy network definition.
According to resent_videoiq.py, the policy network is an instance of models.twod_models.policynet.MobileNetV2.
However, python train_dist.py --dataset mini-sports1m --backbone_net mobilenet_v2 seems to be using a different definition which is from mobilenet.py.
In this case, the CKPT obtained from python train_dist.py --dataset mini-sports1m --backbone_net mobilenet_v2 might not be loaded for python train_dist_ada.py --dataset mini-sports1m --q_init 2 --backbone_net resnet_videoiq successfully.
Could you please provide some guidance on this?