|
@@ -5,10 +5,9 @@ Some sample code (with some pseudo DL framework):
|
|
|
```python
|
|
```python
|
|
|
from cvargparse import GPUParser, ArgFactory, Arg
|
|
from cvargparse import GPUParser, ArgFactory, Arg
|
|
|
|
|
|
|
|
|
|
+from dlframework import Model, LRSchedule, Updater, Iterator, to_gpu
|
|
|
|
|
|
|
|
def main(args):
|
|
def main(args):
|
|
|
-
|
|
|
|
|
-
|
|
|
|
|
data = load_data(args.data, args.labels)
|
|
data = load_data(args.data, args.labels)
|
|
|
model = Model(args.model_weights)
|
|
model = Model(args.model_weights)
|
|
|
# we can select multiple GPUs. use the first GPU for the initial model creation
|
|
# we can select multiple GPUs. use the first GPU for the initial model creation
|
