没想到第一次是这种方式in到代码中，不过以我的性格，自学真的要很久。

学习心得

到时候我完成后提醒你记得把 master 的一些东西 merge 到你的分支中，可能需要学习下~

另外看一下 log 文件夹里面的， 用tensorboard 查看情况,端口号是6006

“/usr/bin/python /usr/local/bin/tensorboard –logdir=./ –port=6006”

跑完了记得对实验重命名，没跑完的 tensorboard log 记得删除

1
2
mv directory_name new_directory_name
rm -rf directory_name

python train_PA.py –data_dir /data/modelnet40/modelnet40_ply_hdf5_2048 –WVAE

add vae loss

论文：PointAugment

目标：在原文基础上，修改，第三部分改为vae（已经做好），增加vaeloss

总结：改为vae（已经做好），困难的部分师兄解决，我的工作就是捋清楚项目结构

add tensorborad（不会）

add wandb，可视化

当前工作，在model.py中增加

  • import wandb

  • wandb.init(project=”PointAugment-VAE”)

  • def backup(self):
    wandb.log({"Train Accuracy":train_acc,"Test Accuracy":test_acc,"CLS Loss": clsLoss.data,"AUG Loss":augLoss.data})
     wandb.log({"input Point Cloud":wandb.Object3D(points[0].transpose(0,1).detach().cpu().numpy()),"Augmented Point Cloud":wandb.Object3D(aug_pc[0].transpose(0,1).detach().cpu().numpy())})

我的工作