sattarov / FedTabDiff

Implementation of the paper: "FedTabDiff: Federated Learning of Diffusion Models for Synthetic Mixed-Type Tabular Data Generation"
https://arxiv.org/abs/2401.06263
MIT License
16 stars 5 forks source link

Update main.ipynb #3

Closed fwtsinghua closed 3 months ago

fwtsinghua commented 3 months ago

fix the device bug.