Disk dataset training - Githubissues

sbdzdz / disco

Disentangled Continual Learning

https://arxiv.org/abs/2312.16731

Apache License 2.0

2 stars 1 forks source link

Disk dataset training #32

Closed sbdzdz closed 9 months ago

sbdzdz commented 9 months ago

Add the option to train from data on disk.

Changes:

add a script for training from data on disk
switch backbone from torchvision to timm
update configs for the baselines
add a script for timing training with vanilla PyTorch
unify SLURM training scripts
update callbacks
simplify the main model