FFJORD复现

复现了任天琦的ffjord文章，记录下

实验环境

• RTX3090
• CentOS Linux release 7.5.1804 (Core) 服务器
• pytorch 2.1.2
• python 3.8
• cuda 12.1

实验参数

1. test1

• python train_cnf.py --data mnist --dims 64,64,64 --strides 1,1,1,1 --num_blocks 2 --layer_type concat --multiscale True --rademacher True --batch_size 1024
  
  

• test2
  python train_cnf.py --data mnist --dims 64,64,64 --strides 1,1,1,1 --num_blocks 2 --layer_type concat --multiscale True --rademacher True

epoch1000：