search

themattinthehatt / behavenet

Toolbox for analyzing behavioral videos and neural activity
https://behavenet.readthedocs.io/
MIT License
57 stars 15 forks source link

remove hardcoded n_epochs #23

Open johnlyzhou opened 3 years ago

johnlyzhou commented 3 years ago

remove the hardcoded conditional requiring n_epochs==200

themattinthehatt commented 3 years ago

@johnlyzhou good catch here. We'll still need to add some additional updates though - filtering by n_epochs ensures that we only plot the loss values at the end of training; not doing so will plot the loss values for all epochs, which we don't want. So you'll have to find the max number of epochs in the dataframe, then filter by that value.