NUS-HPC-AI-Lab / DATM

ICLR 2024, Towards Lossless Dataset Distillation via Difficulty-Aligned Trajectory Matching
https://gzyaftermath.github.io/DATM/
94 stars 6 forks source link

DATM with TESLA #9

Closed silicx closed 6 months ago

silicx commented 6 months ago

Add a TESLA implementation of DATM at ./distill/DATM_tesla.py. This code have been (sparsely) tested on CIFAR10, IPC=1 and 1000, producing comparable performance to the MTT implementation. Hope it could reduce the cost of reproduction and facilitate further research.