updates
This commit is contained in:
parent
8df3601663
commit
bd92457604
3
train.py
3
train.py
@ -320,9 +320,6 @@ def train(cfg,
|
|||||||
return results
|
return results
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
parser = argparse.ArgumentParser()
|
parser = argparse.ArgumentParser()
|
||||||
parser.add_argument('--epochs', type=int, default=100, help='number of epochs')
|
parser.add_argument('--epochs', type=int, default=100, help='number of epochs')
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user