arcee-ai / DAM

30 stars 4 forks source link

change presets, fix random seed, clear GPU memory after training #20

Closed ElliotStein closed 1 month ago

ElliotStein commented 1 month ago

Also added command line arguments to modify dataset size and k (currently just works for precomputed datasets