 Dimitri Korsch
|
26392d602a
some functions were missing
|
4 éve |
 Dimitri Korsch
|
2b1f830114
minor changes in device initialization
|
4 éve |
|
Dimitri Korsch
|
a7f5c1e19b
added loading of default chainercv2 weights
|
4 éve |
|
Dimitri Korsch
|
8e95d9c768
moved prepare function initialization from dataset to model mixin
|
4 éve |
|
Dimitri Korsch
|
5eed48e144
added pre-training command line argument and its handling
|
4 éve |
|
Dimitri Korsch
|
1138a2cb10
some changes in model loading
|
4 éve |
|
Dimitri Korsch
|
844a158386
renamed a method
|
4 éve |
|
Dimitri Korsch
|
145bf0ae60
removed unused imports
|
4 éve |
|
Dimitri Korsch
|
184697051a
refactored finetuner module a bit
|
4 éve |
|
Dimitri Korsch
|
6078c8eeee
new version
|
4 éve |
|
Dimitri Korsch
|
452a7ee5e8
added load_path argument. fixed a minor bug
|
4 éve |
|
Dimitri Korsch
|
0c264c5665
added load_path argument. fixed a minor bug
|
4 éve |
|
Dimitri Korsch
|
f3859d5c61
added evaluation method
|
4 éve |
|
Dimitri Korsch
|
be76b189e0
small fix in the logger init of the finetune parser
|
4 éve |
|
Dimitri Korsch
|
e124478fa1
fixed evaluator progress bar for the case if no_progress was set
|
4 éve |
|
Dimitri Korsch
|
53663bc118
new version
|
4 éve |
|
Dimitri Korsch
|
8d09a95de0
updated code for only_head trainings
|
4 éve |
|
Dimitri Korsch
|
2636439e65
minor bug fixes
|
4 éve |
|
Dimitri Korsch
|
6968bc3bc3
added saving of meta information
|
4 éve |
|
Dimitri Korsch
|
45d9db3bcb
minor bug fixes
|
4 éve |
|
Dimitri Korsch
|
b04e852052
cleaned the trainer class
|
4 éve |
|
Dimitri Korsch
|
d20b137536
fixed some naming bugs
|
4 éve |
|
Dimitri Korsch
|
e545dfeddd
refactored base trainer class: moved some init code to separate methods
|
4 éve |
|
Dimitri Korsch
|
029610e9a6
new version
|
4 éve |
|
Dimitri Korsch
|
dacda213ef
fixed some part input size issues
|
4 éve |
|
Dimitri Korsch
|
488da11c09
added part size handling
|
4 éve |
|
Dimitri Korsch
|
9f143ab908
call method of the separate model classifier is now abstact. this way using this class needs to be defined, how the different models handle the inputs
|
4 éve |
|
Dimitri Korsch
|
fba8601fce
enabled progress bar in the evaluator
|
4 éve |
|
Dimitri Korsch
|
73267a952b
added some augmentation options
|
4 éve |
|
Dimitri Korsch
|
d3ecc94832
new version
|
5 éve |
