PyTorch-MNIST-Tutorial/ ├── LICENSE ├── model.pth ├── model.py ├── README.md ├── test.py ├── train.py ├── utils.py ├── demo.py ├── mnist_train.csv └── mnist_test.csv ...
We implement this work with Ubuntu 18.04, NVIDIA Tesla V100, and CUDA11.8. CUDA_VISIBLE_DEVICES=0 python train_align.py --dataset MS-COCO --model UDIS --epochs 150 -b 16 --loss-type UDIS --freq_record ...
Some results have been hidden because they may be inaccessible to you
Show inaccessible results
Feedback