 Dimitri Korsch
|
5eed48e144
added pre-training command line argument and its handling
|
4 years ago |
 Dimitri Korsch
|
1138a2cb10
some changes in model loading
|
4 years ago |
|
Dimitri Korsch
|
844a158386
renamed a method
|
4 years ago |
|
Dimitri Korsch
|
145bf0ae60
removed unused imports
|
4 years ago |
|
Dimitri Korsch
|
184697051a
refactored finetuner module a bit
|
4 years ago |
|
Dimitri Korsch
|
6078c8eeee
new version
|
4 years ago |
|
Dimitri Korsch
|
452a7ee5e8
added load_path argument. fixed a minor bug
|
4 years ago |
|
Dimitri Korsch
|
0c264c5665
added load_path argument. fixed a minor bug
|
4 years ago |
|
Dimitri Korsch
|
f3859d5c61
added evaluation method
|
4 years ago |
|
Dimitri Korsch
|
be76b189e0
small fix in the logger init of the finetune parser
|
4 years ago |
|
Dimitri Korsch
|
e124478fa1
fixed evaluator progress bar for the case if no_progress was set
|
4 years ago |
|
Dimitri Korsch
|
53663bc118
new version
|
4 years ago |
|
Dimitri Korsch
|
8d09a95de0
updated code for only_head trainings
|
4 years ago |
|
Dimitri Korsch
|
2636439e65
minor bug fixes
|
4 years ago |
|
Dimitri Korsch
|
6968bc3bc3
added saving of meta information
|
4 years ago |
|
Dimitri Korsch
|
45d9db3bcb
minor bug fixes
|
4 years ago |
|
Dimitri Korsch
|
b04e852052
cleaned the trainer class
|
4 years ago |
|
Dimitri Korsch
|
d20b137536
fixed some naming bugs
|
4 years ago |
|
Dimitri Korsch
|
e545dfeddd
refactored base trainer class: moved some init code to separate methods
|
4 years ago |
|
Dimitri Korsch
|
029610e9a6
new version
|
4 years ago |
|
Dimitri Korsch
|
dacda213ef
fixed some part input size issues
|
4 years ago |
|
Dimitri Korsch
|
488da11c09
added part size handling
|
4 years ago |
|
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 years ago |
|
Dimitri Korsch
|
fba8601fce
enabled progress bar in the evaluator
|
4 years ago |
|
Dimitri Korsch
|
73267a952b
added some augmentation options
|
4 years ago |
|
Dimitri Korsch
|
d3ecc94832
new version
|
5 years ago |
|
Dimitri Korsch
|
011c8f6e5b
updated versions in the requirements
|
5 years ago |
|
Dimitri Korsch
|
ddfefc0adc
fixed requirements
|
5 years ago |
|
Dimitri Korsch
|
205a4e3706
updated versions
|
5 years ago |
|
Dimitri Korsch
|
598aec0682
minor weight decay initialization fix: now WD is disabled if value < 0 given
|
5 years ago |
