Merge branch 'master' of github.com:Jittor/JSparse

This commit is contained in:
li-xl 2022-10-20 20:20:32 +08:00
commit 3b001a60b1
1 changed files with 2 additions and 2 deletions

View File

@ -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()