Alexander CW Smith

icon indicating a website link https://scholar.google.com/citations?user=vRhVRVMAAAAJ&hl=en

icon location Charleston, SC icon location I am an Assistant Professor in the Department of Neuroscience at the Medical University of South Carolina. My lab studies the neurobiology of drug addiction.

Results 5 issues of Alexander CW Smith

Release GPU memory after task completion with numba.cuda

12 comment

This PR takes advantage of numba.cuda.close() to release GPU memory upon completion of training, evaluation, or analysis of videos. Very minimal additions, no additional dependencies, no docstring changes. Tested on...

enhancement

License

Hello, Is this project meant to be open-source? If so would you please add an open source license to this repository? Thank you, Alex

Allow resuming training by passing full path to pretrained_model

This commit enables loading a pretrained model from a full model path specified in config.yaml, instead of needing to specify only the model name and not the full path. This...

Added option to evaluate all snapshots, fixed bug when prediction_dec…

I added a use_snapshots boolean parameter to vame.evaluate_model, when set to True it will evaluate all snapshots (similar to DeepLabCut) which allows the user to get a better look at...

Small changes to scheduler threshold

The optimal scheduler threshold parameter can vary greatly between experiments (from .1 to .0001), depending on the order of magnitude of the loss values. In this PR I changed the...