UETAILab / uetai

Custom ML tracking experiment and debugging tools.
https://uetailab.github.io/uetai/
MIT License
15 stars 1 forks source link

Add file upload CLI after training #3

Closed nmd2k closed 2 years ago

nmd2k commented 2 years ago

There is a need to gather artifact and send it into wandb server after training, consider implement that feature.

The module would implement logger.save(path, object and write a temporary file. by call uetai upload file_list.txt, the module should upload every files in file_list.txt and upload that into wandb with same experiment.

gungui98 commented 2 years ago

How to test this feature

logger = SummaryWriter("experiment")
logger.save("model.pth", model)

test code

# in remote wandb we have "experiment/model.pth"
.....