Question about 'Enet (Doc3D + DIW) + Tnet (Doc3D + DIW)' pipeline about paperedge HOT 2 OPEN

thx for the reply
I see, so what I did is correct, firstly I trained the Enet model and obtained the trained Enet model(pt), then I replace the G_ckpt in train.json with the trained Enet model(pt) and train the Tnet model.
I also agree with you that we should not run the two trainers at the same time.

from paperedge.

Basically after training Enet, you should create a new training config json file and point the Enet checkpoint:
https://github.com/cvlab-stonybrook/PaperEdge/blob/main/configs/train.json#L7
https://github.com/cvlab-stonybrook/PaperEdge/blob/main/train.py#L50-L51
then in train_L_step_w the Enet is running with the weights from the previous step.
(sry there are some name confusion for Enet, Gnet, Tnet, Lnet...train and eval have different names...)

train_L_step_w is enough. Do not use 2 engines at the same time...(at least I have not tried)

(sry it took me longer and longer to read all the emails...)

from paperedge.