frutiemax
/

rct_model

Model card Files Files and versions Community

rct_model / train_model.py

Commit History

Use accelerate

f6f5f48

frutiemax commited on Sep 23, 2023

Fix batch size

1618027

frutiemax commited on Sep 23, 2023

Fix training

04d70cd

frutiemax commited on Sep 23, 2023

Use ConditionalUnetModel

88deab4

frutiemax commited on Sep 23, 2023

Training code

d751051

frutiemax commited on Sep 22, 2023