DropIT: DROPPING INTERMEDIATE TENSORS FOR MEMORY-EFFICIENT DNN TRAINING

The dataset used in the paper is not explicitly described, but it is mentioned that the authors used DeiT and ViT models on ImageNet-1k and CIFAR-100 datasets.

Data and Resources

Cite this as

Joya Chen, Kai Xu, Yuhui Wang, Yifei Cheng, Angela Yao (2024). Dataset: DropIT: DROPPING INTERMEDIATE TENSORS FOR MEMORY-EFFICIENT DNN TRAINING. https://doi.org/10.57702/9nn394jv

DOI retrieved: December 2, 2024

Additional Info

Field Value
Created December 2, 2024
Last update December 2, 2024
Author Joya Chen
More Authors
Kai Xu
Yuhui Wang
Yifei Cheng
Angela Yao
Homepage https://github.com/chenjoya/dropit