Merge branch 'master' of github.com:Jittor/JSparse
This commit is contained in:
commit
3b001a60b1
|
@ -180,7 +180,7 @@ def main():
|
|||
jt.flags.use_cuda = 1
|
||||
|
||||
args = parse_args()
|
||||
|
||||
# wget https://shapenet.cs.stanford.edu/media/modelnet40_ply_hdf5_2048.zip and unzip it
|
||||
train_loader = ModelNet40H5(
|
||||
data_root=args.data_root,
|
||||
batch_size=args.batch_size,
|
||||
|
@ -206,4 +206,4 @@ def main():
|
|||
print(f"epoch:{epoch},acc:{acc},lr:{optimizer.lr}")
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
main()
|
||||
|
|
Loading…
Reference in New Issue