search

neufieldrobotics / NeuFlow_v2

Apache License 2.0
92 stars 6 forks source link

It's really difficult to reproduce the results of the paper , can you offer us the detailed training settings? #5

Closed PotentialX closed 2 months ago

PotentialX commented 2 months ago

great work! but i found it is really unstable and difficult to reproduct the result the same as yours.

Study-is-happy commented 2 months ago

val_results.txt We train with batch size 256. Here is the our training record file.

We first start with lr=6e-4, the training code will decrease it to 1e-4 gradually. Then set the lr=6e-4 again, wait it decrease to 1e-4 gradually. Then set the lr=4e-4, wait it decrease to 5e-5. Then set the lr=3e-4, wait it decrease to 5e-5. Then set the lr=2e-4, wait it decrease to 5e-5.

Unfortunately, You have to manually reset the learning rate after some period.

PotentialX commented 2 months ago

bravo!