qinglew / PCN-PyTorch

Implementation of PCN(Point Completion Network) in PyTorch.
141 stars 33 forks source link

some questions about train.py #20

Closed Marigod98 closed 1 year ago

Marigod98 commented 2 years ago

maybe this part should be train_step < 50000? not should be epoch<50000 for epoch in range(1, params.epochs + 1):

hyperparameter alpha

    if train_step < 10000:
        alpha = 0.01
    elif train_step < 20000:
        alpha = 0.1
    elif epoch < 50000:
        alpha = 0.5
    else:
        alpha = 1.0
qinglew commented 1 year ago

Yes, you are right! I have rectified the code.