Kevin Fink
commited on
Commit
·
52ba344
1
Parent(s):
5a4a1bb
dev
Browse files
app.py
CHANGED
|
@@ -168,6 +168,7 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
|
|
| 168 |
return 'SECOND THIRD LOADED'
|
| 169 |
|
| 170 |
except:
|
|
|
|
| 171 |
dataset = load_dataset(dataset_name.strip())
|
| 172 |
train_size = len(dataset['train'])
|
| 173 |
third_size = train_size // 3
|
|
|
|
| 168 |
return 'SECOND THIRD LOADED'
|
| 169 |
|
| 170 |
except:
|
| 171 |
+
print(f"An error occurred: {str(e)}, TB: {traceback.format_exc()}")
|
| 172 |
dataset = load_dataset(dataset_name.strip())
|
| 173 |
train_size = len(dataset['train'])
|
| 174 |
third_size = train_size // 3
|