search

mmSir / SpatioTemporalEntropyModel

Spatio-Temporal Entropy Model (STEM) for end-to-end leaned video compression.
26 stars 1 forks source link

How to train spatio-temporal entropy model #1

Open xuezhongcailian opened 2 years ago

xuezhongcailian commented 2 years ago

When training and testing the spatio-temporal entropy model, is the I-frame training model using a pre-trained model? Another question, args.i_model_path I did not find, is as follows image

mmSir commented 2 years ago

Q1: When training and testing the spatio-temporal entropy model, is the I-frame training model using a pre-trained model? A1: Yes. You can use pretrained model from compressai.

Q2: args.i_model_path can not find A2: It is from input args. example: trainSTEM.py --i-model-path "Pretrained/I/Frame/Model/Path" Then args.i_model_path = "Pretrained/I/Frame/Model/Path"

xuezhongcailian commented 2 years ago

Thanks for your quick reply, but I see the following in the reademe，I can't understand the meaning of --checkpoint image

mmSir commented 2 years ago

Sorry, it is my mistake. --checkpoint is exactly the --i-model-path.

marketprince commented 10 months ago

When training and testing the spatio-temporal entropy model, is the I-frame training model using a pre-trained model? Another question, args.i_model_path I did not find, is as follows image

Have you, senior, successfully run this code completely? I always encounter issues during my run. [crying]

marketprince commented 10 months ago

Why, after completing the training, am I encountering issues with testing?【crying】