Spaces:
Runtime error
Runtime error
geekyrakshit
commited on
Commit
•
60407c7
1
Parent(s):
8ff8f45
updated mirnet
Browse files
enhance_me/mirnet/mirnet.py
CHANGED
@@ -21,7 +21,7 @@ from ..commons import (
|
|
21 |
|
22 |
|
23 |
class MIRNet:
|
24 |
-
def __init__(self, experiment_name
|
25 |
self.experiment_name = experiment_name
|
26 |
if wandb_api_key is not None:
|
27 |
init_wandb("mirnet", experiment_name, wandb_api_key)
|
|
|
21 |
|
22 |
|
23 |
class MIRNet:
|
24 |
+
def __init__(self, experiment_name=None, wandb_api_key=None) -> None:
|
25 |
self.experiment_name = experiment_name
|
26 |
if wandb_api_key is not None:
|
27 |
init_wandb("mirnet", experiment_name, wandb_api_key)
|