Skip to content
Snippets Groups Projects
Commit e8d70f15 authored by Daniel CARRON's avatar Daniel CARRON :b:
Browse files

Moved prediction to lightning

Some DensenetRS-specific code has been removed from predictor.py, it
will have to be re-added directly inside the forward of the DensenetRS model.
parent 9d36fa1f
No related branches found
No related tags found
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment