svjack commited on
Commit
27b8e5e
β€’
1 Parent(s): 51731ac

Upload 2 files

Browse files
BidirectionalTranslation/options/base_options.py CHANGED
@@ -1,4 +1,4 @@
1
- import spaces
2
  import argparse
3
  import os
4
  from util import util
@@ -66,7 +66,7 @@ class BaseOptions():
66
  self.initialized = True
67
  return parser
68
 
69
- @spaces.GPU
70
  def gather_options(self):
71
  """Initialize our parser with basic options (only once)."""
72
  if not self.initialized:
@@ -116,7 +116,7 @@ class BaseOptions():
116
  opt_file.write(message)
117
  opt_file.write('\n')
118
 
119
- @spaces.GPU
120
  def parse(self, model_global_path):
121
  """Parse options, create checkpoints directory suffix, and set up gpu device."""
122
  self.model_global_path = model_global_path
 
1
+ #import spaces
2
  import argparse
3
  import os
4
  from util import util
 
66
  self.initialized = True
67
  return parser
68
 
69
+ #@spaces.GPU
70
  def gather_options(self):
71
  """Initialize our parser with basic options (only once)."""
72
  if not self.initialized:
 
116
  opt_file.write(message)
117
  opt_file.write('\n')
118
 
119
+ #@spaces.GPU
120
  def parse(self, model_global_path):
121
  """Parse options, create checkpoints directory suffix, and set up gpu device."""
122
  self.model_global_path = model_global_path
BidirectionalTranslation/options/test_options.py CHANGED
@@ -1,8 +1,8 @@
1
- import spaces
2
  from .base_options import BaseOptions
3
 
4
  class TestOptions(BaseOptions):
5
- @spaces.GPU
6
  def initialize(self, parser):
7
  BaseOptions.initialize(self, parser)
8
 
 
1
+ #import spaces
2
  from .base_options import BaseOptions
3
 
4
  class TestOptions(BaseOptions):
5
+ #@spaces.GPU
6
  def initialize(self, parser):
7
  BaseOptions.initialize(self, parser)
8