search

mlexchange / mlex_dlsia_segmentation_prototype

Other
3 stars 3 forks source link

call DLSIA’s Trainer and pass dvclive obj to it #20

Closed xiaoyachong closed 8 months ago

xiaoyachong commented 8 months ago

As a new class called Trainer() has been added to DLSIA, we can call this class instead of using the previous train_segmentation() function in train.py. The main changes are summarized as follows:

from dlsia.core.train_scripts import Trainer
from dvclive import Live

dvclive= Live(...)
trainer = Trainer(..., dvclive=dvclive)
net, results = trainer.train_segmentation()