Skip to content

Pytorch v2 upgrade
Compare
Choose a tag to compare
@ThibaultGROUEIX ThibaultGROUEIX released this 23 Nov 15:00
· 110 commits to master since this release
v2.0
25029c6
This commit was created on github.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Verified
Learn about vigilant mode.

Release Atlasnet v2:

  • Upgrade synthax to pytorch v2 and python 3.7 -> installation is much more easy

    • remove Variable
    • '/' -> '//'
    • .data[0] -> .item()
    • transform.scale -> transform.Resize
    • visdom display curve code
    • optimize import packages

  • Added post paper optimizations

    • added learning rate decay scheduler

  • Provide two chamfer distance

    • GPU pytorch code (slow and heavy but simple)
    • GPU pytorchv1 custom cuda layer (fast and light)

  • Update Readme and add some missing comments

Assets 2
Loading