search

valeoai / WaffleIron

Other
39 stars 6 forks source link

Update trainer.py #12

Closed SpeedyGonzales949 closed 3 weeks ago

SpeedyGonzales949 commented 4 weeks ago

Converted Tensor from int32 to int64. This sovles the issue #11